集群混合类

class sklearn.base.ClusterMixin

scikit-learn 中所有聚类估计器的混合类。

• 通过 estimator_type 标签将估计器类型设置为 "clusterer";

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

示例

>>> 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 执行聚类并返回聚类标签。

参数：

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

输入数据。

y忽略

未使用，出于 API 一致性约定而存在。

**kwargs字典

传递给fit的参数。

1.4 版本新增。

返回:

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

聚类标签。