Skip to content

Commit ebaa7b1

Browse files
committed
Update documentation.
1 parent 3d79fc4 commit ebaa7b1

File tree

1 file changed

+28
-2
lines changed

1 file changed

+28
-2
lines changed

hdbscan/hdbscan_.py

Lines changed: 28 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -217,6 +217,15 @@ def hdbscan(X, min_cluster_size=5, min_samples=None, alpha=1.0,
217217
metric parameter.
218218
If metric is "precomputed", X is assumed to be a distance matrix and
219219
must be square.
220+
(default minkowski)
221+
222+
p : int, optional
223+
p value to use if using the minkowski metric. (default 2)
224+
225+
alpha : float, optional
226+
A distance scaling parameter as used in robust single linkage.
227+
See (K. Chaudhuri and S. Dasgupta "Rates of convergence
228+
for the cluster tree."). (default 1.0)
220229
221230
algorithm : string, optional
222231
Exactly which algorithm to use; hdbscan has variants specialised
@@ -238,8 +247,20 @@ def hdbscan(X, min_cluster_size=5, min_samples=None, alpha=1.0,
238247
labels : array [n_samples]
239248
Cluster labels for each point. Noisy samples are given the label -1.
240249
250+
probabilities : array [n_samples]
251+
Cluster membership stringths for each point. Noisy samples are assigned 0.
252+
241253
condensed_tree : record array
242254
The condensed cluster hierarchy used to generate clusters.
255+
256+
single_linkage_tree : array [n_samples - 1, 4]
257+
The single linkage tree produced during clustering in scipy
258+
hierarchical clustering format
259+
(see http://docs.scipy.org/doc/scipy/reference/cluster.hierarchy.html).
260+
261+
min_spanning_tree : array [n_samples - 1, 3]
262+
The minimum spanning as an edgelist. If gen_min_span_tree was False
263+
this will be None.
243264
244265
References
245266
----------
@@ -323,10 +344,15 @@ class HDBSCAN(BaseEstimator, ClusterMixin):
323344
metric parameter.
324345
If metric is "precomputed", X is assumed to be a distance matrix and
325346
must be square.
347+
(default euclidean)
348+
349+
p : int, optional
350+
p value to use if using the minkowski metric. (default None)
326351
327352
alpha : float, optional
328353
A distance scaling parameter as used in robust single linkage.
329-
See (reference paper). (default 1.0)
354+
See (K. Chaudhuri and S. Dasgupta "Rates of convergence
355+
for the cluster tree."). (default 1.0)
330356
331357
332358
gen_min_span_tree: bool, optional
@@ -431,7 +457,7 @@ def condensed_tree_(self):
431457

432458
@property
433459
def single_linkage_tree_(self):
434-
if self._condensed_tree is not None:
460+
if self._single_linkage_tree is not None:
435461
return SingleLinkageTree(self._single_linkage_tree)
436462
else:
437463
warn('No single linkage tree was generated; try running fit first.')

0 commit comments

Comments
 (0)