2014-04-01 157 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'。 –

相关问题