ClusterMixin#

class sklearn.base.ClusterMixin[source]#

scikit-learn 中所有聚类估算器的 Mixin 类。

通过 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)[source]#

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

参数:

Xshape 为 (n_samples, n_features) 的 array-like: Input data.
y被忽略: 未使用，按照惯例为保持 API 一致性而存在。
**kwargsdict: 要传递给 fit 的参数。

1.4 版本新增。

返回:

labels形状为 (n_samples,) 的 ndarray，dtype=np.int64: 聚类标签。

ClusterMixin#

本页