5分pk10app下载 _以预测股票涨跌案例入门基于SVM的机器学习

  • 时间:
  • 浏览:0
  • 来源:欧力资源网 - 专注共享空木白博客分享

    SVM是Support Vector Machine的缩写,中文叫支持向量机,通过它还并能 对样本数据进行分类。以股票为例,SVM能根据若干行态样本数据,把待预测的目标结果划分成“涨”和”跌”有一种,从而实现预测股票涨跌的效果。

1 通过简单案例了解SVM的分类作用

    在Sklearn库里,封装了SVM分类的相关最好的办法,也假使 说,大伙儿儿我不要 了解其中复杂化的算法,即可用它实现基于SVM的分类。通过如下SimpleSVMDemo.py案例,大伙儿儿来看下通过SVM库实现分类的做法,以及相关最好的办法的调用最好的办法。    

1    #!/usr/bin/env python
2    #coding=utf-8
3    import numpy as np
4    import matplotlib.pyplot as plt
5    from sklearn import svm
6    #给出平面上的若干点
7    points = np.r_[[[-1,1],[1.5,1.5],[1.8,0.2],[0.8,0.7],[2.2,2.8],[2.5,3.5],[4,2]]]
8    #按0和1标记成两类
9    typeName = [0,0,0,0,1,1,1]

    在第5行里,大伙儿儿引入了基于SVM的库。在第7行,大伙儿儿定义了若干个点,并在第9行把哪些地方地方点分成了两类,比如[-1,1]点是第一类,而[4,2]是第二类。

    这里请注意,在第7行定义点的后来,是通过np.r_最好的办法,把数据转加进去“列矩阵”,从前做的目的是让数据行态满足fit最好的办法的要求。     

10	#建立模型
11	svmTool = svm.SVC(kernel='linear')
12	svmTool.fit(points,typeName)  #传入参数
13	#确立分类的直线
14	sample = svmTool.coef_[0] #系数
15	slope = -sample[0]/sample[1]  #斜率
16	lineX = np.arange(-2,5,1)#获取-2到5,间距是1的若干数据
17	lineY = slope*lineX-(svmTool.intercept_[0])/sample[1]

  在第11行里,大伙儿儿创建了基于SVM的对象,并指定该SVM模型采用比较常用的“线性核”来实现分类操作。

      在第14行,通过fit训练样本。这里fit最好的办法和后来基于线性回归案例中的fit最好的办法是一样的,只不过这里是基于线性核的相关算法,而后来是基于线性回归的相关算法(比如最小二乘法)。训练完成后,通过第14行和第15行的代码,大伙儿儿得到了能分隔两类样本的直线,包括直线的斜率和截距,并通过第16行和第17行的代码设置了分隔线的若干个点。    

18	#画出划分直线
19	plt.plot(lineX,lineY,color='blue',label='Classified Line')
20	plt.legend(loc='best') #绘制图例
21	plt.scatter(points[:,0],points[:,1],c='R')
22	plt.show()

  计算完成后,大伙儿儿通过第19行的plot最好的办法绘制了分隔线,并在第21行通过scatter最好的办法绘制所有的样本点。可能points是“列矩阵”的数据行态,什么都有是用points[:,0]来获取绘制点的 x坐标,用points[:,1]来获取y坐标,最后是通过第22行的show最好的办法绘制图形。运行上述代码,大伙儿儿能看了如下图13.8的效果,从中大伙儿儿能看了,淡蓝色的边界线能有效地分隔两类样本。

    

    从有一种例子中大伙儿儿能看了,SVM的作用是,根据样本,训练出能划分不同种类数据的边界线,由此实现“分类”的效果。而且,在根据训练样本挑选好边界线的参数后,还能根据其它没有 明确种类样本,计算出它的种类,以此实现“预测”效果。 

2 数据标准化出理

    标准化(normalization)出理 是将行态样本按一定算法进行缩放,让它们落在某个范围比较小的区间,一齐加进去单位限制,让样本数据转加进去无量纲的纯数值。

    在用机器学习最好的办法进行训练时,一般还要进行标准化出理 ,由于 是Sklearn等库封装的一些机器学习算法对样本有一定的要求,可能一些行态值的数量级主次大多数行态值的数量级,可能有行态值主次正态分布,没有 预测结果会不准确。

    还要说明的是,实在在训练前对样本进行了标准化出理 ,改变了样本值,但可能在标准化的过程中是用同一一3个多多多多算法对全部样本进行转换,属于“数据优化”,我不要 对后继的训练起到不好的作用。

    这里大伙儿儿是通过sklearn库提供的preprocessing.scale最好的办法实现标准化,该最好的办法是让行态值减去平均值而且除以标准差。通过如下ScaleDemo.py案例,大伙儿儿实际用下preprocessing.scale最好的办法。     

1	#!/usr/bin/env python
2	#coding=utf-8
3	from sklearn import preprocessing
4	import numpy as np
5	
6	origVal = np.array([[10,5,3],
7	                   [8,6,12],
8	                   [14,7,15]])
9	#计算均值
10	avgOrig = origVal.mean(axis=0)
11	#计算标准差
12	stdOrig=origVal.std(axis=0)
13	#减去均值,除以标准差
14	print((origVal-avgOrig)/stdOrig)
15	scaledVal=preprocessing.scale(origVal)
16	#直接输出preprocessing.scale后的结果
17	print(scaledVal)

  在第6行里,大伙儿儿初始化了一一3个多多多多长宽各为3的矩阵,在第10行,通过mean最好的办法计算了该矩阵的均值,在第12行则通过std最好的办法计算标准差。

      第14行是用原始值减去均值,再除以标准差,在第17行,是直接输出preprocessing.scale的结果。第14行和第17行的输出结果相同,均是下值,从中大伙儿儿验证了标准化的具体做法。    

1	[[-0.26726124 -1.22474487 -1.37281295]
2	 [-1.06904497  0.          0.39223227]
3	 [ 1.33680621  1.22474487  0.98058068]]

3 预测股票涨跌

    在后来的案例中,大伙儿儿用基于SVM的最好的办法,通过一维直线来分类二维的点。据此还并能 进一步推论:通过基于SVM的最好的办法,大伙儿儿还还并能 分类具有多个行态值的样本。

    比如还并能 通过开盘价、收盘价、最高价、最低价和成交量等行态值,用SVM的算法训练出哪些地方地方行态值和股票“涨“和“跌“的关系,即通过行态值划分指定股票“涨”和“跌”的边界,从前得话,一旦输入其它的股票行态数据,即可预测出对应的涨跌状况。在如下的PredictStockBySVM.py案例中,大伙儿儿给出了基于SVM预测股票涨跌的功能。     

1	#!/usr/bin/env python
2	#coding=utf-8
3	import pandas as pd 
4	from sklearn import svm,preprocessing
5	import matplotlib.pyplot as plt
6	origDf=pd.read_csv('D:/stockData/ch13/8038052018-09-012019-05-31.csv',encoding='gbk')
7	df=origDf[['Close', 'Low','Open' ,'Vol','Date']]
8	#diff列表示本日和上日收盘价的差
9	df['diff'] = df["Close"]-df["Close"].shift(1)
10	df['diff'].fillna(0, inplace = True)
11	#up列表示本日是是否是上涨,1表示涨,0表示跌
12	df['up'] = df['diff']   
13	df['up'][df['diff']>0] = 1
14	df['up'][df['diff']<=0] = 0
15	#预测值我不要

说初始化为0
16	df['predictForUp'] = 0

  第6行里,大伙儿儿从指定文件读取了所含股票信息的csv文件,该csv格式的文件实在是从网络数据接口获取得到的,具体做法还并能 参考前面博文。

    从第9行里,大伙儿儿设置了df的diff列为本日收盘价和前日收盘价的差值,通过第12行到第14行的代码,大伙儿儿设置了up列的值,具体是,可能当日股票上涨,即本日收盘价大于前日收盘价,则up值是1,反之可能当日股票下跌,up值则为0。

    在第16行里,大伙儿儿在df对象里新建了表示预测结果的predictForUp列,该列的值我不要 说都设置为0,在后继的代码里,将根据预测结果填充这列的值。    

17	#目标值是真实的涨跌状况
18	target = df['up']
19	length=len(df)
20	trainNum=int(length*0.8)
21	predictNum=length-trainNum
22	#挑选指定列作为行态列
23	feature=df[['Close', 'High', 'Low','Open' ,'Volume']]
24	#标准化出理

行态值
25	feature=preprocessing.scale(feature)

  在第18行里,大伙儿儿设置训练目标值是表示涨跌状况的up列,在第20行,设置了训练集的数量是总量的80%,在第23行则设置了训练的行态值,请注意这里加进去了日期有一种不相关的列,而且,在第25行,对行态值进行了标准化出理 。    

26	#训练集的行态值和目标值
27	featureTrain=feature[1:trainNum-1]
28	targetTrain=target[1:trainNum-1]
29	svmTool = svm.SVC(kernel='liner')
80	svmTool.fit(featureTrain,targetTrain)

  在第27行和第28行里,大伙儿儿通过截取指定行的最好的办法,得到了行态值和目标值的训练集,在第26行里,以线性核的最好的办法创建了SVM分类器对象svmTool。

     在第80行里,通过fit最好的办法,用行态值和目标值的训练集训练svmTool分类对象。从上文里大伙儿儿可能看了,训练所用的行态值是开盘收盘价、最高最低价和成交量,训练所用的目标值是描述涨跌状况的up列。在训练完成后,svmTool对象中就所含了能划分股票涨跌的相关参数。

31	predictedIndex=trainNum
32	#逐行预测测试集
33	while predictedIndex<length:
34	    testFeature=feature[predictedIndex:predictedIndex+1]            
35	    predictForUp=svmTool.predict(testFeature)    
36	    df.ix[predictedIndex,'predictForUp']=predictForUp    
37	    predictedIndex = predictedIndex+1

    在第33行的while循环里,大伙儿儿通过predictedIndex索引值,依次遍历测试集。

    在遍历过程中,通过第35行的predict最好的办法,用训练好的svmTool分类器,逐行预测测试集中的股票涨跌状况,并在第36行里,把预测结果设置到df对象的predictForUp列中。      

38	#该对象只所含预测数据,即只所含测试集
39	dfWithPredicted = df[trainNum:length]
40	#刚开始

绘图,创建一一3个多多多多子图
41	figure = plt.figure()
42	#创建子图     
43	(axClose, axUpOrDown) = figure.subplots(2, sharex=True)
44	dfWithPredicted['Close'].plot(ax=axClose)
45	dfWithPredicted['predictForUp'].plot(ax=axUpOrDown,color="red", label='Predicted Data')
46	dfWithPredicted['up'].plot(ax=axUpOrDown,color="blue",label='Real Data')
47	plt.legend(loc='best') #绘制图例
48	#设置x轴坐标标签和旋转淬硬层

49	major_index=dfWithPredicted.index[dfWithPredicted.index%2==0]
80	major_xtics=dfWithPredicted['Date'][dfWithPredicted.index%2==0]
51	plt.xticks(major_index,major_xtics)
52	plt.setp(plt.gca().get_xticklabels(), rotation=80) 
53	plt.title("通过SVM预测803805的涨跌状况")
54	plt.rcParams['font.sans-serif']=['SimHei']
55	plt.show()

  可能在后来的代码里,大伙儿儿只设置测试集的predictForUp列,并没有 设置训练集的该列数据,什么都有在第39行里,用切片的手段,把测试集数据放置到dfWithPredicted对象中,请注意这里切片的起始和刚开始 值是测试集的起始和刚开始 索引值。至此完成了数据准备工作,在后来的代码里,大伙儿儿将用matplotlib库刚开始 绘图。

    在第43行里,大伙儿儿通过subplots最好的办法设置了一一3个多多多多子图,并通过sharex=True让这些一3个多多多多子图的x轴具有相同的刻度和标签。在第44行代码里,在axClose子图中,大伙儿儿用plot最好的办法绘制了收盘价的走势。在第45行代码里,在axUpOrDown子图中,大伙儿儿绘制了预测到的涨跌状况,而在第46行里,还是在axUpOrDown子图里,绘制了哪些地方地方天的股票真实的涨跌状况。

    在第49行到第52行的代码里,大伙儿儿设置了x标签的文字以及旋转淬硬层 ,从前做的目的是让标签文字看上去不至于太密集。在第53行里,大伙儿儿设置了中文标题,可能要显示中文,什么都有还要第54行的代码,最后在55行通过show最好的办法展示了图片。运行上述代码,能看了如下图所示的效果。

    

    其中上图展示了收盘价,下图的淡蓝色线条表示真实的涨跌状况,0表示跌,1表示上涨,而红色则表示预测后的结果。

4 结论

     对比一下,虽有偏差,但大体相符。综上所述,本案例是数学淬硬层 ,演示了通过SVM分类的做法,包括可能划分行态值和目标值,如可对样本数据进行标准化出理 ,如可用训练数据训练SVM,还有如可用训练后的结果预测分类结果。 

5 总结和版权说明

    本文是给tcp连接员加财商系列,后来还有两篇博文

    本文的内容即将出书,在出版的书里,是用股票案例和大伙儿儿讲述Python入门时的知识点,敬请期待

    

    有不少老外转载和要我转载我的博文,被委托人感到十分荣幸,这也是被委托人不断写博文的动力。关于本文的版权有如下统一的说明,抱歉就不逐一回复了。

    1 本文可转载,我不要 告知,转载时请用链接的最好的办法,给出原文出处,别简单地通过文本最好的办法给出,一齐写明原作者是hsm_computer。

    2 在转载时,请原文转载 ,谢绝洗稿。而且被委托人保留追究法律责任的权利。