ClassifierMixin#
- class sklearn.base.ClassifierMixin[源代码]#
Scikit-learn中所有分类器的Mixin类。
此mixin定义了以下功能:
将估计器类型设置为
"classifier"
通过estimator_type
标签;score
默认为accuracy_score
.强制执行该
fit
需要y
被传递通过requires_y
标签,这是通过设置分类器类型标签来完成的。
阅读更多的 User Guide .
示例
>>> import numpy as np >>> from sklearn.base import BaseEstimator, ClassifierMixin >>> # Mixin classes should always be on the left-hand side for a correct MRO >>> class MyEstimator(ClassifierMixin, BaseEstimator): ... def __init__(self, *, param=1): ... self.param = param ... def fit(self, X, y=None): ... self.is_fitted_ = True ... return self ... def predict(self, X): ... return np.full(shape=X.shape[0], fill_value=self.param) >>> estimator = MyEstimator(param=1) >>> X = np.array([[1, 2], [2, 3], [3, 4]]) >>> y = np.array([1, 0, 1]) >>> estimator.fit(X, y).predict(X) array([1, 1, 1]) >>> estimator.score(X, y) 0.66...