0

GridSearchCVScikitを使用して、パイプラインのパラメーターを検索しています。私は自分のコードを機能させましたが、追加したい場合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 に追加する適切な方法は何ですか?

4

1 に答える 1

2

シンプル -パラメータ名が間違っています:

class_weight : 辞書、辞書のリスト、「balanced」、「balanced_subsample」または None、オプション

于 2016-05-04T07:58:00.170 に答える