Skip to content

Commit 178802b

Browse files
committed
Added some comments to the code.
1 parent ef8e40d commit 178802b

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

hdbscan/hdbscan_.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,7 @@
1919
from sklearn.externals import six
2020
from warnings import warn
2121

22+
#Try and work around older sklearn api
2223
try:
2324
from sklearn.utils import check_array
2425
except ImportError:
@@ -46,7 +47,9 @@
4647

4748

4849
def _tree_to_labels(X, single_linkage_tree, min_cluster_size=10):
49-
50+
""" Converts a pretrained tree and cluster size into a
51+
set of labels and probabilities.
52+
"""
5053
condensed_tree = condense_tree(single_linkage_tree,
5154
min_cluster_size)
5255
stability_dict = compute_stability(condensed_tree)
@@ -72,6 +75,9 @@ def _hdbscan_generic(X, min_samples=5, alpha=1.0,
7275

7376
min_spanning_tree = mst_linkage_core(mutual_reachability_)
7477

78+
#mst_linkage_core does not generate a full minimal spanning tree
79+
#If a tree is required then we must build the edges from the information
80+
#returned by mst_linkage_core (i.e. just the order of points to be merged)
7581
if gen_min_span_tree:
7682
result_min_span_tree = min_spanning_tree.copy()
7783
for index, row in enumerate(result_min_span_tree[1:], 1):

0 commit comments

Comments
 (0)