python - 在 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 构造函数参数。注意双下划线。