ClusterMixin

class sklearn.base.ClusterMixin

Mixin类用于scikit-learn中的所有聚类估计器。

• _estimator_type 类属性默认为 "clusterer" ；

• fit_predict 方法返回与每个样本相关的聚类标签。

Examples

>>> import numpy as np
>>> from sklearn.base import BaseEstimator, ClusterMixin
>>> class MyClusterer(ClusterMixin, BaseEstimator):
...     def fit(self, X, y=None):
...         self.labels_ = np.ones(shape=(len(X),), dtype=np.int64)
...         return self
>>> X = [[1, 2], [2, 3], [3, 4]]
>>> MyClusterer().fit_predict(X)
array([1, 1, 1])

fit_predict(X, y=None, **kwargs)

执行对 X 的聚类并返回聚类标签。

Parameters:

X形状为 (n_samples, n_features) 的类数组

输入数据。

y忽略

未使用，为保持API一致性而存在。

**kwargs字典

传递给 fit 的参数。

Added in version 1.4.

Returns:

labels形状为 (n_samples,)，dtype=np.int64 的 ndarray

聚类标签。