ignore self neighbors from the start

TutteInstitute · Dec 24, 2024 · ca7b56c · ca7b56c
1 parent fc49ae9
commit ca7b56c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/fast_hdbscan/boruvka.py b/fast_hdbscan/boruvka.py
@@ -306,4 +306,4 @@ def parallel_boruvka(tree, min_samples=10, sample_weights=None):
 
     edges = np.vstack(edges)
     edges[:, 2] = np.sqrt(edges.T[2])
-    return edges, neighbors, np.sqrt(core_distances)
+    return edges, neighbors[:, 1:], np.sqrt(core_distances)
diff --git a/fast_hdbscan/core_graph.py b/fast_hdbscan/core_graph.py
@@ -23,7 +23,7 @@ def knn_mst_union(neighbors, core_distances, min_spanning_tree, lens_values):
         children = graph[point]
         parent_lens = lens_values[point]
         parent_dist = core_distances[point]
-        for child in neighbors[point, 1:]:
+        for child in neighbors[point]:
             if child < 0:
                 continue
             children[child] = (