网格搜索正在进行中

Question

我正在使用Scikit中的GridSearchCV搜索管道中的参数。我让我的代码正常工作，但如果我想添加class_weights，我就碰壁了。

from sklearn.pipeline import Pipeline
RFC = RandomForestClassifier()
PCA = PCA()
pipe = Pipeline(steps=[('PCA', PCA), ('RFC', RFC)])
param_dict = {'RFC__n_estimators': [100,150],
              'RFC__class_weights': [{0:1,1:2},{0:1,1:4}],
              'PCA__n_components': [60,80]}
from sklearn.grid_search import GridSearchCV             
estimator = GridSearchCV(pipe, param_dict, scoring='roc_auc')
estimator.fit(X_train, y_train)

如何将此参数添加到GridSearch的正确方法是什么？

Answer 1

Simple-您得到了错误的参数名称：

class_weight：dict，dicts列表，"balanced"、"balanced_subsample"或None，可选