python-sklearn中SVM调参中画学习曲线

2020-03-28  本文已影响0人  丙吉

脑子经常短路,就把看到的随手给记录下,以防后续查看使用:

score = []

gamma_range = np.logspace(-10, 1, 50)

for i in gamma_range:

clf = SVC(kernel='rbf', gamma = i, cache_size=5000).fit(Xtrain, Ytrain)

score.append(slf.score(Xtest, Ytest))

print(max(score), gamma_range[score.index(max(score))])

plt.plot(gamma_range, score)

plt.show()

上一篇下一篇

猜你喜欢

热点阅读