Spaces:
Build error
Build error
n_jobs deprecated
Browse files- src/Surveyor.py +2 -2
src/Surveyor.py
CHANGED
@@ -521,7 +521,7 @@ class Surveyor:
|
|
521 |
optimal_k = self.model.calculate_optimal_k(' '.join(abs_lines), k_max=10)
|
522 |
# Perform kmean clustering
|
523 |
|
524 |
-
clustering_model = KMeans(n_clusters=optimal_k, n_init=20
|
525 |
# clustering_model = AgglomerativeClustering(n_clusters=optimal_k, affinity='cosine', linkage='average') #, affinity='cosine', linkage='average', distance_threshold=0.4)
|
526 |
clustering_model.fit(corpus_embeddings)
|
527 |
cluster_assignment = clustering_model.labels_
|
@@ -561,7 +561,7 @@ class Surveyor:
|
|
561 |
optimal_k = self.model.calculate_optimal_k(' '.join(abs_lines), k_max=10)
|
562 |
# Perform kmean clustering
|
563 |
|
564 |
-
clustering_model = KMeans(n_clusters=optimal_k, n_init=20
|
565 |
# clustering_model = AgglomerativeClustering(n_clusters=optimal_k, affinity='cosine', linkage='average') #, affinity='cosine', linkage='average', distance_threshold=0.4)
|
566 |
clustering_model.fit(corpus_embeddings)
|
567 |
cluster_assignment = clustering_model.labels_
|
|
|
521 |
optimal_k = self.model.calculate_optimal_k(' '.join(abs_lines), k_max=10)
|
522 |
# Perform kmean clustering
|
523 |
|
524 |
+
clustering_model = KMeans(n_clusters=optimal_k, n_init=20)
|
525 |
# clustering_model = AgglomerativeClustering(n_clusters=optimal_k, affinity='cosine', linkage='average') #, affinity='cosine', linkage='average', distance_threshold=0.4)
|
526 |
clustering_model.fit(corpus_embeddings)
|
527 |
cluster_assignment = clustering_model.labels_
|
|
|
561 |
optimal_k = self.model.calculate_optimal_k(' '.join(abs_lines), k_max=10)
|
562 |
# Perform kmean clustering
|
563 |
|
564 |
+
clustering_model = KMeans(n_clusters=optimal_k, n_init=20)
|
565 |
# clustering_model = AgglomerativeClustering(n_clusters=optimal_k, affinity='cosine', linkage='average') #, affinity='cosine', linkage='average', distance_threshold=0.4)
|
566 |
clustering_model.fit(corpus_embeddings)
|
567 |
cluster_assignment = clustering_model.labels_
|