华量杯-股票预测, keras+LSTM
共 2629字,需浏览 6分钟
·
2020-08-03 02:45
向AI转型的程序员都关注了这个号???
机器学习AI算法工程 公众号:datayx
对每日A股数据取样,提取出包括时间序列类型指标,非时间序列类型指标以及股票分类属性指标;利用所提供的指标来预测T -1日的个股收益。
一、数据预处理
代码:clean.py
二、利用LSTM模型
1. 安装keras框架
Keras安装之前,需要先安装好numpy,scipy。下面是在windows下的安装。
(1)安装pip
https://pypi.python.org/pypi/pip#downloads
下载对应版本的pip。如"pip-9.0.1.tar.gz (md5, pgp)"
然后解压,进入到pip-9.0.1这个目录中,运行下面的代码安装
python setup.py install
重启,使环境变量生效
(2)安装numpy
注意,不能用pip install numpy的方式安装,会缺少依赖的库。采用下面的方法:
下载numpy‑1.11.3+mkl‑cp27‑cp27m‑win_amd64.whl,(由于我的python版本是2.7.9,是windows 64位)下载的地址为:
http://www.lfd.uci.edu/~gohlke/pythonlibs/#numpy
下载好之后,进入到numpy‑1.11.3+mkl‑cp27‑cp27m‑win_amd64.whl所在目录,运行下面的命令安装:
pip install numpy‑1.11.3+mkl‑cp27‑cp27m‑win_amd64.whl
(3)安装scipy
注意,不能用pip install scipy的方式安装,会报下面的错:
正确的做法是,采用下面的方法进行安装:
首先,下载scipy‑0.19.0‑cp27‑cp27m‑win_amd64.whl,(由于我的python版本是2.7.9,是windows 64位)下载的地址为:
http://www.lfd.uci.edu/~gohlke/pythonlibs/#scipy
下载好之后,进入到scipy‑0.19.0‑cp27‑cp27m‑win_amd64.whl所在目录,运行下面的命令安装:
pip install scipy‑0.19.0‑cp27‑cp27m‑win_amd64.whl
(4)安装keras 运行下面的命令:
pip install keras
现在keras己经安装好了。接下来就可以用Keras提供的LSTM进行训练了!
2. 训练,测试,评估
在运行代码前需要把keras的backend改一下,改成theano,而不用tensorflow。因为theano在keras安装时己经安装好了,而tensorflow还要重新安装。首先找到keras.json文件,在下面的目录:
C:\Users\zhangyanni\.keras\keras.json
然后把下面"backend": "tensorflow" 中的tensorflow改成theano
接下来,就可以运行predict.py了
代码: predict.py
代码+数据集 获取:
关注微信公众号 datayx 然后回复 股票 即可获取。
AI项目体验地址 https://loveai.tech
阅读过本文的人还看了以下文章:
基于40万表格数据集TableBank,用MaskRCNN做表格检测
《深度学习入门:基于Python的理论与实现》高清中文PDF+源码
2019最新《PyTorch自然语言处理》英、中文版PDF+源码
《21个项目玩转深度学习:基于TensorFlow的实践详解》完整版PDF+附书代码
PyTorch深度学习快速实战入门《pytorch-handbook》
【下载】豆瓣评分8.1,《机器学习实战:基于Scikit-Learn和TensorFlow》
李沐大神开源《动手学深度学习》,加州伯克利深度学习(2019春)教材
【Keras】完整实现‘交通标志’分类、‘票据’分类两个项目,让你掌握深度学习图像分类
如何利用全新的决策树集成级联结构gcForest做特征工程并打分?
Machine Learning Yearning 中文翻译稿
斯坦福CS230官方指南:CNN、RNN及使用技巧速查(打印收藏)
中科院Kaggle全球文本匹配竞赛华人第1名团队-深度学习与特征工程
不断更新资源
深度学习、机器学习、数据分析、python
搜索公众号添加: datayx
机大数据技术与机器学习工程
搜索公众号添加: datanlp
长按图片,识别二维码