将pipeline与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

查看此处了解更多详细信息。