python – sklearn pipeline – 在管道中应用多项式特征转换后应用样本权重

我想应用样本权重,同时使用来自sklearn的管道,该管道应该进行特征转换,例如多项式,然后应用回归量,例如ExtraTrees.

我在以下两个示例中使用以下包：

from sklearn.ensemble import ExtraTreesRegressor
import numpy as np
from sklearn.pipeline import Pipeline
from sklearn.preprocessing import PolynomialFeatures

只要我单独转换功能并在之后生成和训练模型,一切都很顺利：

#Feature generation
X = np.random.rand(200,4)
Y = np.random.rand(200)

#Feature transformation
poly = PolynomialFeatures(degree=2)
poly.fit_transform(X)

#Model generation and fit
clf = ExtraTreesRegressor(n_estimators=5,max_depth = 3)
weights = [1]*100 + [2]*100
clf.fit(X,Y,weights)

但是在管道中执行它不起作用：

#Pipeline generation
pipe = Pipeline([('poly2',PolynomialFeatures(degree=2)),('ExtraTrees',ExtraTreesRegressor(n_estimators=5,max_depth = 3))])

#Feature generation
X = np.random.rand(200,4)
Y = np.random.rand(200)

#Fitting model
clf = pipe
weights = [1]*100 + [2]*100
clf.fit(X,weights)

我收到以下错误：TypeError：fit()最多需要3个参数(给定4个)
在这个简单的例子中,修改代码没有问题,但是当我想在我的实际代码中对我的真实数据运行几个不同的测试时,能够使用管道和样本权重

解决方法

For this,it enables setting parameters of the various steps using their names and the parameter name separated by a ‘__’,as in the example below.

所以说的是,尝试将最后一行更改为：

clf.fit(X,**{'ExtraTrees__sample_weight': weights})

This is a good example如何使用管道中的参数.

（编辑：商洛站长网）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!