パイプラインのグリッド検索

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

シンプル - あなたは間違っparameter nameを持っている：

class_weight：辞書、dictsのリスト、「バランス」、「balanced_subsample」またはなし、オプションの