群集混合類#
- 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])