首页 文章

在Sklearn中使用GridSearchCV的OneVsRestClassification

提问于
浏览 次
5

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

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

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

python scikit-learn svm

1 回答

  • 4

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

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

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

    回复于

相关问题