Skip to content

(陆续更新)重新整理过的基于机器学习的股票价格预测算法,里面包含了基本的回测系统以及各种不同的机器学习算法的股票价格预测,包含:LSTM算法、Prophet算法、AutoARIMA、朴素贝叶斯、SVM等

Notifications You must be signed in to change notification settings

GERMINO-LiuHe/A-stock-prediction-algorithm-based-on-machine-learning

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

14 Commits
 
 
 
 
 
 
 
 

Repository files navigation

A-stock-prediction-algorithm-based-on-machine-learning

(陆续更新)重新整理过的基于机器学习的股票价格预测算法,里面包含了基本的回测系统以及各种不同的机器学习算法的股票价格预测,包含:LSTM算法、Prophet算法、AutoARIMA、朴素贝叶斯、SVM等

3-15

机器学习算法/kNN.py kNN算法,即k-近邻算法,KNN是通过测量不同特征值之间的距离进行分类。它的思路是:如果一个样本在特征空间中的k个最相似(即特征空间中最邻近)的样本中的大多数属于某一个类别,则该样本也属于这个类别,其中K通常是不大于20的整数。KNN算法中,所选择的邻居都是已经正确分类的对象。该方法在定类决策上只依据最邻近的一个或者几个样本的类别来决定待分样本所属的类别。

3-13

机器学习算法/SVM.py 支持向量机(Support Vector Machine),通过SVM算法对该数据集第N+1日的收盘价相对于第N日的收盘价涨跌情况进行分析,若第N+1日的收盘价相对于第N日收盘价为涨,则记为1,;反之,记为0。由于SVM算法为监督学习算法,因此在训练的过程中算法会根据实际的涨跌情况对学习算法进行修正。对于测试集中的部分,预测每一个交易日的股价涨跌情况,并以真实涨跌情况进行测算,计算在测试集中算法预测的正确率。SVM算法包中提供了多种核函数,在这里提供了三种。

机器学习算法/Prophet.py Prophet算法,也称为预言家算法。Prophet属于时间序列模型。相对于ARIMA模型,Prophet模型的优势在于可以不考虑缺失值的填充问题。另外,Prophet模型的拟合速度远比ARIMA模型更快。

3-12

机器学习算法/AutoARIMA.py 差分回归积分移动平均模型(Autoregressive Integrated Moving Average Model),通过给定d,p,q值,实现在给定的范围内自动寻找最优解,主要给定两种方法,一是一次性预测(即单一预测,使用训练集的数据直接给出所有测试集天数的预测值),另一种是按天预测,(即每日回归,每一个预测期的单位数据都用之前的数据作为训练集进行预测)

机器学习算法/LSTM.py 长短神经网络,一种特殊的循环神经网络,用于处理和预测时间序列中间隔和延迟相对较长的重要事件。LSTM通过三个这样的本结构来实现信息的保护和控制。这三个门分别输入门、遗忘门和输出门。由于该神经网络随着时间变化具有数据的遗忘性,所以一次性预测与按天预测并没有较大的不同,但是该神经网络算法比较复杂,感兴趣的可以参考别的文献,在这里只是实现了基本的功能。

3-11

回测.py 提供了一个基本的回测demo,实现股票基本的买入卖出,考虑佣金和印花税

传统技术面分析算法/Moving_Average.py 移动平均算法,通过移动窗口的方式进行股票预测,传入窗口长度参数可以进行预测,支持tushare接口直接获取数据并即时进行预测

传统技术面分析算法/Relative_strength_index.py 相对强度指数,相对强弱指数RSI是根据一定时期内上涨点数和涨跌点数之和的比率制作出的一种技术曲线。返回计算期内的相对强度指数计算情况

2020-3-10

创建项目,README文件

About

(陆续更新)重新整理过的基于机器学习的股票价格预测算法,里面包含了基本的回测系统以及各种不同的机器学习算法的股票价格预测,包含:LSTM算法、Prophet算法、AutoARIMA、朴素贝叶斯、SVM等

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages