在scikit-learn管道步骤中添加fit_params

时间:2019-05-08 16:08:06

标签: scikit-learn

在管道上调用fit_params时,我不想指定scikit学习管道的.fit()之一,而是要让其中一个管道步骤生成参数值,特别是:

Pipeline(steps=
    [
        ('step1', 
        CustomTransformer()), <-- this ought to generate sample_weights for step2
        ('step2', 
        LogisticRegression(...))
    ]
)

可以做到吗?

0 个答案:

没有答案