在Sklearn中使用GridSearchCV的OneVsRestClassification

Question

当我尝试通过以下方式搜索rbf内核SVM中的最佳C和gamma时：

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

它返回错误说C不是OneVsRestClassifier的参数。那么使用多类SVM实现参数网格搜索的正确方法是什么？

Answer 1

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

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

名称estimator对应于OneVsRestClassifier构造函数参数。请注意双下划线。