문제

When I tried to search the optimal C and gamma in rbf kernel SVM by:

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

It returns the error says C is not the parameter of OneVsRestClassifier. What is the proper way to achieve the grid search on the parameters with multiclass SVM then?

도움이 되었습니까?

해결책

Grid search uses a custom nested attribute syntax for this:

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

The name estimator corresponds to the OneVsRestClassifier constructor parameter. Note the double underscores.

라이센스 : CC-BY-SA ~와 함께 속성
제휴하지 않습니다 StackOverflow
scroll top