python – Sklearn中使用GridSearchCV的OneVsRestClassification

前端之家收集整理的这篇文章主要介绍了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构造函数参数.请注意双下划线.

原文链接:https://www.f2er.com/python/439747.html

猜你在找的Python相关文章