Kuautli commited on
Commit
73db4fc
·
verified ·
1 Parent(s): e187fca

Update clustering.py

Browse files
Files changed (1) hide show
  1. clustering.py +3 -1
clustering.py CHANGED
@@ -615,6 +615,7 @@ def perform_clustering(
615
 
616
  if not threshold_values:
617
  threshold_values = np.round(np.linspace(min_eps, max_eps, n), 2)
 
618
  # threshold_values = np.linspace(min_eps, max_eps, n)
619
 
620
  cluster_assignments = {}
@@ -624,7 +625,7 @@ def perform_clustering(
624
  most_similar_comments = {}
625
 
626
  for distance_threshold in threshold_values:
627
- distance_threshold = round(distance_threshold, 6)
628
  clustering = AgglomerativeClustering(
629
  n_clusters=None,
630
  distance_threshold=distance_threshold,
@@ -694,6 +695,7 @@ def build_sankey_data(
694
  comments = []
695
 
696
  threshold_values = sorted(cluster_assignments.keys())
 
697
  valid_clusters = {}
698
 
699
  for threshold in threshold_values:
 
615
 
616
  if not threshold_values:
617
  threshold_values = np.round(np.linspace(min_eps, max_eps, n), 2)
618
+ print(threshold_values)
619
  # threshold_values = np.linspace(min_eps, max_eps, n)
620
 
621
  cluster_assignments = {}
 
625
  most_similar_comments = {}
626
 
627
  for distance_threshold in threshold_values:
628
+ print(distance_threshold)
629
  clustering = AgglomerativeClustering(
630
  n_clusters=None,
631
  distance_threshold=distance_threshold,
 
695
  comments = []
696
 
697
  threshold_values = sorted(cluster_assignments.keys())
698
+ print("build_sankey_data", threshold_values)
699
  valid_clusters = {}
700
 
701
  for threshold in threshold_values: