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_
|