Fix 264. Incorrect affinity parameter (NVIDIA#265)

Fix NVIDIA#264. Scikit-learn is not expecting anymore the "affinity parameter" Signed-off-by: Miguel Martínez <[email protected]> Signed-off-by: Vinay Raman <[email protected]>
vinay-raman · Nov 12, 2024 · 73c015a · 73c015a
1 parent ffb2774
commit 73c015a
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/tutorials/synthetic-retrieval-evaluation/DeDup.py b/tutorials/synthetic-retrieval-evaluation/DeDup.py
@@ -32,7 +32,6 @@ def clustering(self, embeddings, threshold=0.075):
 
         agg_clustering = AgglomerativeClustering(
             n_clusters=None,
-            affinity="precomputed",
             linkage="complete",
             distance_threshold=threshold,
         )