ClusterMixin

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 一致性约定而存在。

**kwargsdict

要传递给 fit 的参数。

版本 1.4 中新增。

返回:

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

聚类标签。