Find Cluster -> Finding clusters

UKPLab · Dec 12, 2023 · a8ae010 · a8ae010
1 parent 4436a51
commit a8ae010
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sentence_transformers/util.py b/sentence_transformers/util.py
@@ -361,7 +361,7 @@ def community_detection(embeddings, threshold=0.75, min_community_size=10, batch
     min_community_size = min(min_community_size, len(embeddings))
     sort_max_size = min(max(2 * min_community_size, 50), len(embeddings))
 
-    for start_idx in tqdm(range(0, len(embeddings), batch_size), desc="Find Cluster", disable=not show_progress_bar):
+    for start_idx in tqdm(range(0, len(embeddings), batch_size), desc="Finding clusters", disable=not show_progress_bar):
         # Compute cosine similarity scores
         cos_scores = cos_sim(embeddings[start_idx:start_idx + batch_size], embeddings)