python - 将流水线与 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了解更多详情。