使用sklearn管道

Question

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

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

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

我得到像

这样的东西

  

方法适合缺失

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

Answer 1

如果您只是在管道的中间状态转换数据，则不需要实现拟合方法，因此您只需返回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

查看here了解更多详情。