在Sklearn中使用GridSearchCV进行OneVsRest分类

5

当我尝试通过以下方式在rbf核支持向量机中寻找最优的Cgamma时:

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

返回的错误提示为C不是OneVsRestClassifier的参数。那么,在多类SVM参数网格搜索中,正确的方法是什么?

1个回答

4

网格搜索使用自定义的嵌套属性语法:

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

estimator这个名称对应于OneVsRestClassifier构造器的参数。请注意双下划线。


@ChuNan:根据问题的不同,使用较小的模型可能更好,并且它会学习得更快。如果速度太慢,请尝试在OvR包装器上设置n_jobs=-1 - Fred Foo

网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接