【发布时间】:2020-02-22 05:52:33
【问题描述】:
实现我的第一个管道,返回此错误:“无效参数 randomforestclassifier for estimator Pipeline”
skf = StratifiedKFold(n_splits=3, shuffle=True, random_state=SEED)
classifier = Pipeline([
('vectorizer', CountVectorizer(max_features=8000, ngram_range=(1, 5))),
('clf', RandomForestClassifier(n_estimators=10, random_state=15, n_jobs=-1))])
min_samples_leaf = [5, 6, 7, 8]
max_features = [0.3, 0.4, 0.5, 0.6, 0.7]
rfc_params = {'randomforestclassifier__min_samples_leaf': min_samples_leaf,
'randomforestclassifier__max_features':max_features}
class_grid = GridSearchCV(classifier, param_grid = rfc_params,
cv=skf, scoring='roc_auc', n_jobs=-1)
class_grid.fit(X_text, y_text)
【问题讨论】:
标签: python scikit-learn pipeline random-forest