@@ -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