ClassifierMixin#

class sklearn.base.ClassifierMixin[source]#

scikit-learn 中所有分类器的 Mixin 类。

此 mixin 定义了以下功能：

通过 estimator_type 标签将估计器类型设置为 "classifier"；
score 方法，默认使用 accuracy_score。
通过 requires_y 标签（通过设置分类器类型标签完成）强制要求 fit 方法必须传入 y。

在用户指南中阅读更多内容。

示例

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