在Sklearn中使用GridSearchCV的OneVsRestClassification

Question

当我尝试通过以下方式搜索最佳C和gammarbf内核SVM时:

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