2014-04-01 68 views
3

當我試圖通過搜索最優Cgamma在RBF核SVM:OneVsRestClassification與GridSearchCV在Sklearn

params = dict(C = C_range, gamma = gamma_range) 
clf = GridSearchCV(OneVsRestClassifier(SVC()),params, cv = 5) 

它返回錯誤說C不是OneVsRestClassifier參數。然後用多類SVM實現網格搜索參數的正確方法是什麼?

回答

3

網格搜索使用自定義的嵌套屬性的語法如下:

params = dict(estimator__C=C_range, estimator__gamma=gamma_range) 

名稱estimator相當於OneVsRestClassifier構造函數的參數。請注意雙下劃線。

+0

@ChuNan:取決於問題,它可能會更好,它會學習更小的模型。如果速度太慢,嘗試在OvR包裝上設置'n_jobs = -1'。 –

相關問題