python – Sklearn中使用GridSearchCV的OneVsRestClassification

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

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

它返回错误,表示C不是OneVsRestClassifier的参数.那么使用多类SVM实现参数网格搜索的正确方法是什么?
最佳答案
网格搜索使用自定义嵌套属性语法:

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

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

dawei

【声明】:唐山站长网内容转载自互联网,其相关言论仅代表作者个人观点绝非权威,不代表本站立场。如您发现内容存在版权问题,请提交相关链接至邮箱:bqsm@foxmail.com,我们将及时予以处理。