华量杯-股票预测, keras+LSTM

2020-08-04 10:21:18 浏览数 (1)

对每日A股数据取样,提取出包括时间序列类型指标,非时间序列类型指标以及股票分类属性指标;利用所提供的指标来预测T -1日的个股收益。

一、数据预处理

代码:clean.py

二、利用LSTM模型

1. 安装keras框架

Keras安装之前,需要先安装好numpy,scipy。下面是在windows下的安装。

(1)安装pip

代码语言:javascript复制
https://pypi.python.org/pypi/pip#downloads

下载对应版本的pip。如"pip-9.0.1.tar.gz (md5, pgp)"

然后解压,进入到pip-9.0.1这个目录中,运行下面的代码安装

代码语言:javascript复制
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位)下载的地址为:

代码语言:javascript复制
http://www.lfd.uci.edu/~gohlke/pythonlibs/#numpy

下载好之后,进入到numpy‑1.11.3 mkl‑cp27‑cp27m‑win_amd64.whl所在目录,运行下面的命令安装:

代码语言:javascript复制
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位)下载的地址为:

代码语言:javascript复制
http://www.lfd.uci.edu/~gohlke/pythonlibs/#scipy

下载好之后,进入到scipy‑0.19.0‑cp27‑cp27m‑win_amd64.whl所在目录,运行下面的命令安装:

代码语言:javascript复制
pip install scipy‑0.19.0‑cp27‑cp27m‑win_amd64.whl

(4)安装keras 运行下面的命令:

代码语言:javascript复制
pip install keras

现在keras己经安装好了。接下来就可以用Keras提供的LSTM进行训练了!

2. 训练,测试,评估

在运行代码前需要把keras的backend改一下,改成theano,而不用tensorflow。因为theano在keras安装时己经安装好了,而tensorflow还要重新安装。首先找到keras.json文件,在下面的目录:

代码语言:javascript复制
C:Userszhangyanni.keraskeras.json

然后把下面"backend": "tensorflow" 中的tensorflow改成theano

接下来,就可以运行predict.py了

代码: predict.py

代码 数据集 获取:

关注微信公众号 datayx 然后回复 股票 即可获取。

AI项目体验地址 https://loveai.tech

0 人点赞