Spaces:

wilmerags
/

tweet-snest

Build error

wilmerags commited on Nov 24, 2021

Commit

3bc7bbb

1 Parent(s): 629cfc4

fix: Balance min_cluster_size for less topics

Files changed (1) hide show

app.py CHANGED Viewed

@@ -70,7 +70,7 @@ def generate_plot(
         embeddings = embed_text(df, model)
     # encoded_labels = encode_labels(labels)
     cluster = hdbscan.HDBSCAN(
-        min_cluster_size=2,
         metric='euclidean',
         cluster_selection_method='eom'
     ).fit(embeddings)

         embeddings = embed_text(df, model)
     # encoded_labels = encode_labels(labels)
     cluster = hdbscan.HDBSCAN(
+        min_cluster_size=5,
         metric='euclidean',
         cluster_selection_method='eom'
     ).fit(embeddings)