Skip to content

Commit f53e4bb

Browse files
authored
Add min_span_tree warning per #279
1 parent 3fc2b6f commit f53e4bb

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

hdbscan/hdbscan_.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -209,6 +209,11 @@ def _hdbscan_prims_kdtree(X, min_samples=5, alpha=1.0,
209209

210210
# Convert edge list into standard hierarchical clustering format
211211
single_linkage_tree = label(min_spanning_tree)
212+
213+
if gen_min_span_tree:
214+
warn('Cannot generate Minimum Spanning Tree; '
215+
'the implemented Prim\'s does not produce '
216+
'the full minimum spanning tree ', UserWarning)
212217

213218
return single_linkage_tree, None
214219

@@ -240,6 +245,11 @@ def _hdbscan_prims_balltree(X, min_samples=5, alpha=1.0,
240245
:]
241246
# Convert edge list into standard hierarchical clustering format
242247
single_linkage_tree = label(min_spanning_tree)
248+
249+
if gen_min_span_tree:
250+
warn('Cannot generate Minimum Spanning Tree; '
251+
'the implemented Prim\'s does not produce '
252+
'the full minimum spanning tree ', UserWarning)
243253

244254
return single_linkage_tree, None
245255

0 commit comments

Comments
 (0)