Spaces:

de-Rodrigo
/

Embeddings

Sleeping

de-Rodrigo commited on Oct 27

Commit

5e1576b

1 Parent(s): 8951751

Fix Bug

Files changed (1) hide show

app.py CHANGED Viewed

@@ -644,8 +644,8 @@ def compute_global_regression(df_combined, embedding_cols, tsne_params, df_f1, r
     # elbow_k = best_k
     kmeans_opt = KMeans(n_clusters=elbow_k, random_state=42, n_init=10)
-    labels_opt = kmeans_opt.fit_predict(X)
-    silhouette_opt = silhouette_score(X, labels_opt)
     centers = kmeans_opt.cluster_centers_
     radii, densities, volumes = [], [], []

     # elbow_k = best_k
     kmeans_opt = KMeans(n_clusters=elbow_k, random_state=42, n_init=10)
+    labels_opt = kmeans_opt.fit_predict(reduced_real)
+    silhouette_opt = silhouette_score(reduced_real, labels_opt)
     centers = kmeans_opt.cluster_centers_
     radii, densities, volumes = [], [], []