前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >华量杯-股票预测, keras+LSTM

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

作者头像
机器学习AI算法工程
发布2020-08-04 10:21:18
8180
发布2020-08-04 10:21:18
举报

对每日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:\Users\zhangyanni\.keras\keras.json

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

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

代码: predict.py

代码+数据集 获取:

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

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

本文参与 腾讯云自媒体分享计划,分享自微信公众号。
原始发表:2020-08-02,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 机器学习AI算法工程 微信公众号,前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体分享计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • 一、数据预处理
  • 二、利用LSTM模型
  • 1. 安装keras框架
  • 代码+数据集 获取:
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档