ClusterMixin

class sklearn.base.ClusterMixin

scikit-learn中所有集群估计器的Mixin类。

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

• 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 .

Added in version 1.4.

返回:

labels形状的nd数组（n_samples，），dype =np.int64

集群标签。