python svr回归_建模支持向量回归（SVR）与线性回归

我是一个有点新的建模技术，我试图比较SVR和线性回归。我使用了f(x)=5x+10线性函数来生成训练和测试数据集。到目前为止，我已经编写了以下代码片段：import csv

import pandas as pd

import numpy as np

import matplotlib.pyplot as plt

from sklearn.linear_model import LinearRegression

with open('test.csv', 'r') as f1:

train_dataframe = pd.read_csv(f1)

X_train = train_dataframe.iloc[:30,(0)]

y_train = train_dataframe.iloc[:30,(1)]

with open('test.csv','r') as f2:

test_dataframe = pd.read_csv(f2)

X_test = test_dataframe.iloc[30:,(0)]

y_test = test_dataframe.iloc[30:,(1)]

svr = svm.SVR(kernel="rbf", gamma=0.1)

log = LinearRegression()

svr.fit(X_train.reshape(-1,1),y_train)

log.fit(X_train.reshape(-1,1), y_train)

predSVR = svr.predict(X_test.reshape(-1,1))

predLog = log.predict(X_test.reshape(-1,1))

plt.plot(X_test, y_test, label='true data')

plt.plot(X_test, predSVR, 'co', label='SVR')

plt.plot(X_test, predLog, 'mo', label='LogReg')

plt.legend()

plt.show()

如图所示，线性回归效果良好，但支持向量机的预测精度较差。

如果你有任何解决这个问题的建议，请告诉我。

谢谢