SVR三种不同的核预测情况对比

import numpy as np
import matplotlib.pyplot as plt
from sklearn.svm import SVR
from sklearn.metrics import mean_squared_error


if __name__ == '__main__':
    N = 100      # 准备创建的数据个数
    np.random.seed(0)
    x = np.sort(np.random.uniform(0, 6, N))
    y = 2*np.sin(x) + 0.1*np.random.rand(N)
    x = x.reshape(-1, 1)
    
    clfs = [SVR(C=1, kernel='linear'),
            SVR(C=1, kernel='rbf', gamma=0.1),
            SVR(C=1, kernel='poly', degree=3)
            ]
    names = ['linear', 'rbf', 'poly']
    colors = ['g', 'm', 'b']
    plt.figure(figsize=(10, 8))
    plt.plot(x, y, 'ro', markersize=6)
    for i, model in enumerate(clfs):
        model.fit(x, y)
        y_hat = model.predict(x)
        if i == 0:
            print('linear的均方误差是:%.3f' % mean_squared_error(y_hat, y))
        el