Skip to content

Commit 0cd4685

Browse files
authored
Update prediction.py
1 parent aea2239 commit 0cd4685

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

hdbscan/prediction.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -108,7 +108,7 @@ def __init__(self, data, condensed_tree, min_samples,
108108
raw_condensed_tree = condensed_tree._raw_tree
109109

110110
self.cluster_map = {c: n for n, c in enumerate(sorted(list(selected_clusters)))}
111-
self.reverse_cluster_map = {n: c for c, n in cluster_map.items()}
111+
self.reverse_cluster_map = {n: c for c, n in self.cluster_map.items()}
112112

113113
self.cluster_tree = raw_condensed_tree[raw_condensed_tree['child_size']
114114
> 1]

0 commit comments

Comments
 (0)