赞
踩
本文分享基于GRU实现一维和多维预测的python代码,数据来源于tushare导出的股价数据,笔者的tushareID是469251,这篇文章也是笔者使用tushare模块需要完成的小任务,第一次发文,如有纰漏,还请指正。如有涉及到产权问题,请联系删除。
如果有不了解GRU的读者可以在这个网站查看介绍及相关原理:循环神经网络
参考部分代码的原链接:LSTM进阶:使用LSTM进行多维多步的时间序列预测
使用tushare导出股票数据。'YOUR TOKEN'是个人接口,注册tushare后达到一定积分后可凭TOKEN码直接使用tushare下载金融、财务等相关数据。
- ts.set_token('YOUR TOKEN')
- pro = ts.pro_api('YOUR TOKEN')
- df = pro.query('daily', ts_code='600000.SH', start_date='20180101', end_date='20211220')
-
- writer = pd.ExcelWriter('./SH600000.xlsx')
- df.to_excel(writer,sheet_name='SH600000')
- writer.save()
导出股票代码为SH600000,时间为20180101-20211220期间的数据,数据预览如下
基于GRU使用100天的收盘价预测后25天的收盘价,使用最后125天的数据做测试,之前的数据训练网络,代码如下:
- import numpy as np
- import tensorflow as tf
- from tensorflow.keras.layers import Dropout, Dense, LSTM,SimpleRNN,GRU
- import matplotlib.pyplot as plt
- from keras.models import load_model
- import os
- from sklearn.p
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。