2016-04-02 17 views
3

私は複数ラベルの問題があります。私はSVMでOneVsRestClassifierを使用します。今私はGridSearchCVでパラメータを調整したい。試しましたOneVsRestClassifierでGridSearchCVを実行するには?

GridSearchCV(estimator=OneVsRestClassifier(svm.SVC(probability=True)), param_grid=dict(C=Cs), 
         n_jobs=-1) 

さまざまなエラー情報を返します。

OneVsRestClassifierでGridSearchCVを実行するにはどうすればいいですか?多分私はパイプラインを行うべきですか?しかし、OneVsRestClassifierとSVMとの関係は、パイプライン関数の期待通りではないようです。


以下のコードも試しました。しかし、私はsvm.SVCにパラメータを渡すことはできません。

parameters = {'kernel':('linear', 'rbf'), 'C':[1, 10]} 
svr = OneVsRestClassifier(svm.SVC(probability=True)) 
clf = grid_search.GridSearchCV(svr, parameters) 
clf.fit(X_ptrain, y_ptrain) 
+1

を[これはあなたを助けるべきです](http://stackoverflow.com/questions/12632992/gridsearch-for-アステスト・イン・ア・ワンストレス分類器) – Kevin

答えて

0

下図のようにあなたがSVCのパラメータを参照するために推定器を使用することができます。

parameters = {'estimator__kernel':('linear', 'rbf'), 'estimator__C':[1, 10]} 
svr = OneVsRestClassifier(svm.SVC(probability=True)) 
clf = grid_search.GridSearchCV(svr, parameters) 
clf.fit(X_ptrain, y_ptrain) 
関連する問題