在sklearn中使用管道

Question

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

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

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

我得到的东西

method fit is missing

我是不是在课程定义上遗漏了什么？在

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

查看here了解更多详细信息。在