python - Sklearn set_params 正好接受 1 个参数？

Question

我正在尝试使用 SkLearn Bayes classification .

 gnb = GaussianNB()
 gnb.set_params('sigma__0.2')
 gnb.fit(np.transpose([xn, yn]), y)

但是我得到:

set_params() takes exactly 1 argument (2 given)

---

现在我尝试使用这段代码:

gnb = GaussianNB()
arr = np.zeros((len(labs),len(y)))
arr.fill(sigma)
gnb.set_params(sigma_ = arr)

并得到:

ValueError: Invalid parameter sigma_ for estimator GaussianNB

是不是参数名或值错误？

Answer 1

我只是偶然发现了这个，所以这里有一个字典中的多个参数的解决方案:

from sklearn import svm
params_svm = {"kernel":"rbf", "C":0.1, "gamma":0.1, "class_weight":"auto"}
clf = svm.SVC()
clf.set_params(**params_svm)