sidphbot commited on
Commit
c952ddd
1 Parent(s): 27a82f7

n_jobs deprecated

Browse files
Files changed (1) hide show
  1. 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, n_jobs=-1)
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, n_jobs=-1)
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_