BaseEstimator#

class sklearn.base.BaseEstimator[source]#

scikit-learn 中所有估计器的基类。

继承此类的对象将获得以下默认实现：

用于 GridSearchCV 及相关对象的参数设置与获取；
在终端和 IDE 中显示的文本和 HTML 表示；
估计器序列化；
参数验证；
数据验证；
特征名称验证。

请参阅用户指南以了解更多信息。

备注

所有估计器都应在其 __init__ 方法中，将所有可在类级别设置的参数指定为显式关键字参数（不使用 *args 或 **kwargs）。

示例

>>> import numpy as np
>>> from sklearn.base import BaseEstimator
>>> class MyEstimator(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=2)
>>> estimator.get_params()
{'param': 2}
>>> X = np.array([[1, 2], [2, 3], [3, 4]])
>>> y = np.array([1, 0, 1])
>>> estimator.fit(X, y).predict(X)
array([2, 2, 2])
>>> estimator.set_params(param=3).fit(X, y).predict(X)
array([3, 3, 3])