Fix 264. Incorrect affinity parameter (#265)

Fix #264. Scikit-learn is not expecting anymore the "affinity parameter" Signed-off-by: Miguel Martínez <[email protected]>
NVIDIA · Sep 25, 2024 · 10aaa8a · 10aaa8a
1 parent eb2225b
commit 10aaa8a
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,
         )