【发布时间】:2018-07-23 07:49:21
【问题描述】:
我正在使用重要的特征选择来实现一个管道,然后使用相同的特征来训练我的随机森林分类器。以下是我的代码。
m = ExtraTreesClassifier(n_estimators = 10)
m.fit(train_cv_x,train_cv_y)
sel = SelectFromModel(m, prefit=True)
X_new = sel.transform(train_cv_x)
clf = RandomForestClassifier(5000)
model = Pipeline([('m', m),('sel', sel),('X_new', X_new),('clf', clf),])
params = {'clf__max_features': ['auto', 'sqrt', 'log2']}
gs = GridSearchCV(model, params)
gs.fit(train_cv_x,train_cv_y)
所以X_new是通过SelectFromModel和sel.transform选择的新功能。然后我想使用所选的新功能来训练我的 RF。
我收到以下错误:
所有中间步骤都应该是变形金刚并实施适合和 变换,ExtraTreesClassifier ...
【问题讨论】:
标签: python machine-learning scikit-learn feature-selection