是否可以执行GridSearchCV
(以获得最好的 SVM 的 C)并sample_weight
使用 scikit-learn 指定?
这是我的代码和我遇到的错误:
gs = GridSearchCV(
svm.SVC(C=1),
[{
'kernel': ['linear'],
'C': [.1, 1, 10],
'probability': [True],
'sample_weight': sw_train,
}]
)
gs.fit(Xtrain, ytrain)
>> ValueError:估计器 SVC 的参数 sample_weight 无效
编辑:我通过获取最新的 scikit-learn 版本并使用以下内容解决了这个问题:
gs.fit(Xtrain, ytrain, fit_params={'sample_weight': sw_train})