Spaces:

de-Rodrigo
/

Embeddings

Sleeping

de-Rodrigo commited on Oct 27

Commit

ee88baf

1 Parent(s): f7e8a0b

Fix Bug

Files changed (1) hide show

app.py CHANGED Viewed

@@ -630,7 +630,7 @@ def compute_global_regression(df_combined, embedding_cols, tsne_params, df_f1, r
     for k in K:
         kmeans = KMeans(n_clusters=k, random_state=42)
         kmeans.fit(reduced_real)
-        labels_opt = kmeans_opt.fit_predict(X)
         silohuettes_test.append(silhouette_score(X, labels_opt))
     kl = KneeLocator(K, silohuettes_test, curve="convex", direction="decreasing")

     for k in K:
         kmeans = KMeans(n_clusters=k, random_state=42)
         kmeans.fit(reduced_real)
+        labels_opt = kmeans.fit_predict(X)
         silohuettes_test.append(silhouette_score(X, labels_opt))
     kl = KneeLocator(K, silohuettes_test, curve="convex", direction="decreasing")