利用sklearn对球磨机数据进行SVR处理

#-*- coding:utf-8 -*-
from sklearn import svm
import scipy.io as sio
import numpy as np
import sklearn.metrics as ms
data_x=sio.loadmat(u'F:/球磨机数据/对比实验/1_data/train_x.mat')
train_x=data_x['train_x']
data_ctl=sio.loadmat(u'F:/球磨机数据/对比实验/1_data/train_ctly.mat')
train_y=np.array(data_ctl['train_y']).ravel()
clf=svm.SVR(kernel='rbf', C=1e3, gamma=0.1)
clf.fit(train_x,train_y)
test_x=sio.loadmat(u'F:/球磨机数据/对比实验/1_data/test_x.mat')['test_x']
test_y=np.array(sio.loadmat(u'F:/球磨机数据/对比实验/1_data/test_ctly.mat')['test_y']).ravel()
#print(map(lambda x,y:x-y,clf.predict(test_x),test_y))
error=ms.mean_squared_error(test_y,clf.predict(test_x))
print(error)