Kuautli commited on
Commit
e22fec3
·
verified ·
1 Parent(s): 7f3e076

Update clustering.py

Browse files
Files changed (1) hide show
  1. clustering.py +2 -1
clustering.py CHANGED
@@ -627,7 +627,7 @@ def perform_clustering(
627
 
628
  if not threshold_values:
629
  threshold_values = np.round(np.linspace(min_eps, max_eps, n), 6)
630
- log_message(threshold_values)
631
  # threshold_values = np.linspace(min_eps, max_eps, n)
632
 
633
  cluster_assignments = {}
@@ -637,6 +637,7 @@ def perform_clustering(
637
  most_similar_comments = {}
638
 
639
  for distance_threshold in threshold_values:
 
640
  log_message(distance_threshold)
641
  clustering = AgglomerativeClustering(
642
  n_clusters=None,
 
627
 
628
  if not threshold_values:
629
  threshold_values = np.round(np.linspace(min_eps, max_eps, n), 6)
630
+ log_message(f"perform_clustering {threshold_values}")
631
  # threshold_values = np.linspace(min_eps, max_eps, n)
632
 
633
  cluster_assignments = {}
 
637
  most_similar_comments = {}
638
 
639
  for distance_threshold in threshold_values:
640
+ distance_threshold = round(distance_threshold, 6)
641
  log_message(distance_threshold)
642
  clustering = AgglomerativeClustering(
643
  n_clusters=None,