如何在Python sklearn中修复SVR图

时间:2018-04-11 14:35:09

标签: python matplotlib machine-learning scikit-learn svm

我正在尝试将SVR模型拟合到我的数据集中,并使用Python中的Sklearn查看绘图。

from sklearn.svm import SVR
#Load Data
X_train_Occ = pd.DataFrame(X_train['occupancy'])
Y_train_Occ = Y_train
#Rescale
sc_X = StandardScaler()
sc_Y = StandardScaler()
X_train_Occ_scaled = sc_X.fit_transform(X_train_Occ)
Y_train_Occ_scaled = sc_Y.fit_transform(Y_train_Occ.reshape(-1, 1))

regressor = SVR(kernel ='rbf')
regressor.fit(X_train_Occ_scaled, Y_train_Occ_scaled)

我将数据加载到X和Y数据帧中并进行缩放。 见下图:

enter image description here

然后我得到以下输出:

SVR(C=1.0, cache_size=200, coef0=0.0, degree=3, epsilon=0.1, gamma='auto', kernel='rbf', max_iter=-1, shrinking=True, tol=0.001, verbose=False)

然后我尝试用这个来显示回归的结果:

plt.scatter(X_train_Occ_scaled, Y_train_Occ_scaled, color = 'red')
plt.plot(X_train_Occ_scaled, regressor.predict(X_train_Occ_scaled), color = 'blue')
plt.title('Occupancy vs Flow (SVR)')
plt.xlabel('Occupancy')
plt.ylabel('Flow')
plt.show()

其中给出了以下情节:

enter image description here

模型是否过度拟合数据?或者代码有问题吗?

我正在关注此处的代码: http://scikit-learn.org/stable/auto_examples/svm/plot_svm_regression.html

我试图绘制最适合模型的线,而不是每个点的线。

2 个答案:

答案 0 :(得分:1)

如前所述,解决方案是首先通过自变量对数据进行排序,然后将数据拟合到模型中并预测结果。

enter image description here

答案 1 :(得分:0)

请勿使用plt.plot,因为所有数据都是随机排序的。使用plt.scatter或将数据从最小到最大排在第一位