如何使用GridSearchCV进行OneVsRestClassifier？

Question

我有一个多标签的问题。我和支持向量机一起使用OneVsRestClassifier。现在，我希望通过GridSearchCV来调优参数。我试过了

GridSearchCV(estimator=OneVsRestClassifier(svm.SVC(probability=True)), param_grid=dict(C=Cs),
                      n_jobs=-1)

它返回各种错误信息。

如何使用GridSearchCV进行OneVsRestClassifier？也许我该做个管道？然而，OneVsRestClassifier与支持向量机之间的关系似乎并不像所期望的流水线函数那样。

我也尝试了下面的代码。但是，我不能将参数传递到svm.SVC。

parameters = {'kernel':('linear', 'rbf'), 'C':[1, 10]}
svr = OneVsRestClassifier(svm.SVC(probability=True))
clf = grid_search.GridSearchCV(svr, parameters)
clf.fit(X_ptrain, y_ptrain)

Answer 1

您可以使用估值器引用SVC的参数，如下所示：

parameters = {'estimator__kernel':('linear', 'rbf'), 'estimator__C':[1, 10]}
svr = OneVsRestClassifier(svm.SVC(probability=True))
clf = grid_search.GridSearchCV(svr, parameters)
clf.fit(X_ptrain, y_ptrain)