【发布时间】:2016-09-30 18:05:31
【问题描述】:
train.sort_values(by=['mass'], ascending=True, inplace=True)
x = train['mass']
y = train['pa']
# Fit regression model
svr_rbf = SVR(kernel='rbf', C=1e3, gamma=0.1)
svr_lin = SVR(kernel='linear', C=1e3)
svr_poly = SVR(kernel='poly', C=1e3, degree=2)
x_train = x.reshape(x.shape[0], 1)
x = x_train
y_rbf = svr_rbf.fit(x, y).predict(x)
y_lin = svr_lin.fit(x, y).predict(x)
y_poly = svr_poly.fit(x, y).predict(x)
# look at the results
plt.scatter(x, y, c='k', label='data')
plt.hold('on')
plt.plot(x, y_rbf, c='g', label='RBF model')
plt.plot(x, y_lin, c='r', label='Linear model')
plt.plot(x, y_poly, c='b', label='Polynomial model')
plt.xlabel('data')
plt.ylabel('target')
plt.title('Support Vector Regression')
plt.legend()
plt.show()
代码复制自http://scikit-learn.org/stable/auto_examples/svm/plot_svm_regression.html。 而我改变的只是数据集。不知道怎么回事。
【问题讨论】:
-
请更好地格式化您问题中的代码。
标签: python machine-learning scikit-learn svm