使用sklearn管道

Question

我正在尝试定义一个量化器,用于sklearn中的Pipeline/GridSearchCV.定义如下

class Quantizer(base.BaseEstimator, base.TransformerMixin):
    def __init__(self):

    def transform(X, y=None):
      some code 

我得到了类似的东西

方法适合缺失

我在班级的定义中遗漏了什么吗？

Answer 1

如果您只是在管道的中间状态中转换数据,则不需要实现fit方法,因此您只需返回self:

class Quantizer(base.BaseEstimator, base.TransformerMixin):
    def __init__(self):

    def transform(self, X, y=None):
      # some code 

    def fit(self, X, y=None, **fit_params):
      return self

看看这里了解更多细节.