gary yong gary yong - 8 months ago 67
Python Question

I get a linear regression using the SVR by python scikit-learn when the data is not linear

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 =, y).predict(x)
y_lin =, y).predict(x)
y_poly =, y).predict(x)

# look at the results
plt.scatter(x, y, c='k', label='data')
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.title('Support Vector Regression')

The code is copied from
And what I change is only the dataset. I do not know what is the matter.


Most likely has to do with the scale of your data. You are using the same penalty hyper-parameter as they are in the example, but your y values are orders of magnitude greater. Thus, the SVR algorithm will favor simplicity over accuracy since your penalty for error is now small compared to your y values. You need to increase C to say 1e6 (or normalize your y values).

You can see that this is the case if you make C very small in their example code, say C=.00001. Then you get the same kind of results that you are getting in your code.

(More on the algorithm here.)

As a side note, a huge part of Machine Learning practice is hyper-parameter tuning. This is a good example of how even a good base model can yield bad results if provided with the wrong hyper-parameters.