Update clustering.py

clustering.py

@@ -598,11 +598,12 @@ def create_3d_umap_plot(data):
     fig.show()
 
 
-def perform_clustering(data, min_eps, max_eps=0.95, n=5, embeddings_col="embeddings"):
+def perform_clustering(data, min_eps, max_eps=0.95, n=5, threshold_values=None, embeddings_col="embeddings"):
 
-    embeddings_matrix = np.array(data[embeddings_col].tolist())
-    # threshold_values = np.round(np.linspace(min_eps, max_eps, n), 2)
-    threshold_values = np.linspace(min_eps, max_eps, n)
+    if not threshold_values:
+        embeddings_matrix = np.array(data[embeddings_col].tolist())
+        # threshold_values = np.round(np.linspace(min_eps, max_eps, n), 2)
+        threshold_values = np.linspace(min_eps, max_eps, n)
 
     cluster_assignments = {}
     cluster_counts = {}