1. # coding=utf-8
    2. from data_load import load_data
    3. import numpy as np
    4. from sklearn.svm import SVR
    5. from sklearn.model_selection import train_test_split  # 这里是引用了交叉验证
    7. X,y = load_data()
    8. # 训练数据
    9. X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2)
    10. clf = SVR(kernel='rbf',C=1e3, epsilon=0.2,gamma=0.01)
    11. # model =  clf.fit(X_train, y_train)
    12. for i in range(2):
    13.     clf.fit(X_train, y_train)
    14. # 预测
    15. y_pred = clf.predict(X_test)