利用SVM回归器预测车辆数量,数据集为traiffic_data.txt,可以在https://archive.ics.uci.edu/ml/dataset下面,数据中存为:星期、时间、对手棒队、棒球比赛是否正在继续、通过汽车数量,输入测试数据,预测车辆数量。
代码封装成一个类,测试单个数据要注意编码器中进行转换,这一点花了我较长时间,读者也要注意。
__author__ = 'Administrator'
# SVM regressor to estimate traffic
import numpy as np
from sklearn import preprocessing
from sklearn.svm import SVR
class MyTraffic:
X=[];
y=[];
#为训练用
transform_x=[];
transform_y=[] #预处理的数据。
label_encoder=[] #解码器为处理数据做准备
m=0;n=0;fignum=1 #为画图用
paramsList=[] #为评估模型用
def __init__(self,params={'kernel': 'rbf', 'C': 10.0, 'epsilon': 0.2} ):
self.moduleparams=params
self.module=SVR(**params)