分类器Mixin#
- class sklearn.base.ClassifierMixin[source]#
scikit-learn中所有分类器的Mixin类。
此Mixin定义以下功能
通过
estimator_type
标签将估计器类型设置为"classifier"
;默认使用
accuracy_score
的score
方法。强制要求
fit
方法通过requires_y
标签传递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...
图库示例#
__sklearn_is_fitted__ 作为开发者 API
__sklearn_is_fitted__ 作为开发者 API
元数据路由