xiaoping2367 / hs300_stock_predict

该项目用于对沪深300股票的预测,包括股票下载,数据清洗,LSTM 模型的训练,测试,以及实时预测

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

hs300_stock_predict

该项目用于对hs300股票的预测,包括股票下载,数据清洗,LSTM 模型的训练,测试,以及实时预测。

文件构成

1、data_utils.py 用于股票数据下载,清洗,合并等。该文件共有9个函数。

get_stock_data(code, date1, date2, filename, length=-1)
该函数用于下载股票数据,保存开、高、收、低、量、涨跌幅等6维数据。
由于用的tushare接口,因此只能下载最近两年的数据。(从新浪网易财经的数据爬虫接口后续开放)
共有5个参数
code是需要下载的股票代码,如000001是平安银行的股票代码,输入'000001'既下载平安银行的股票数据。
date1是开始日期,格式如"2019-01-03",date2是结束日期,格式同上。
filename是存放数据的目录,如"D:\data"
length是筛选股票长度,默认为-1,既对下载的股票数据长度上不做筛选,如果人为指定长度如200,既会将时间长度200以下的数据剔除不予以保存。

get_hs300_data(date1, date2, filename)
该函数用于下载沪深300指数数据,参数格式同get_stock_data

update_stock_data(filename)
该函数将股票数据从本地文件的最后日期更新至当日,filename是指定的单只股票路径名称,如"d:\data\000001.csv"

get_data_len(file_path)
该函数过去单只股票的时间长度,file_path是单只股票路径名称,如"d:\data\000001.csv"

select_stock_data(file1, file2, date1, date2)
该函数对已经再本地的文件按照日期筛选,date1是开始数据,date2是结束数据,file1是源文件夹,file2是筛选日期后文件存放的文件夹

crop_stock(df, date)
该函数暂时不使用

fill_stock_data(target, sfile, tfile)
该函数按照沪深300指数的时间长度来对个股停盘数据进行填充,填充为该股上一交易日的数据。该函数是对所选文件夹下所有文件进行处理。
注意,如果开始日期是属于停牌的,那么该段停牌将不会被填充,后续会有更新。
target为参照股票,一般选择同时间段的沪深300指数文件,sfile为原文件夹,tfile为填充完要存放文件夹。

merge_stock_data(filename, tfile)
该函数将多个文件按序合并为一个文件,如讲沪深300只个股文件合并为一个总文件,方便后续模型输入。
filename是需要合并的文件夹路径,tfile是存放合并后文件的文件夹路径。

quchong(file)
该函数暂不使用。

2、dataprocess.py 用于训练数据的处理,归一化等,模型的输入都由该文件的接口输出提供。

get_train_data(batch_size=args.batch_size, time_step=args.time_step)
该函数用于处理训练数据,参数默认,有配置文件给定。该函数返回五个变量:batch_index训练集分批处理的批次, val_index验证集批次, train_x_1, 训练集输入,train_y_1, 训练集标签,val_x, y验证集输入,val_y验证集标签
备注:由于整个数据处理是对多只股票合成的总文件处理,所以在时间步长迭代添加时须在各自股票时间长度内完成,因此,需要在配置文件中指定股票长度。

get_test_data(time_step=args.time_step)
该函数用于处理测试集数据,返回两个变量:test_x测试集输入, test_y测试集标签。

get_update_data(time_step=args.time_step)
该函数将更新数据加历史数据的前time_step-1拼接,用于整批处理,如2019 1-3月数据和2018.12的数据拼接,返回拼接后的train_x, train_y
get_predict_data(file_name)
该函数完成下载实时股票数据,与之前的数据拼接后返回输入x。有一个参数需要填充,file_name既要预测的单只股票文件。

3、config.py 配置文件,所有接口内超参数,路径等,在该文件修改,即可在全局生效。

4、lstm_model 模型,包括训练,微调,测试,及预测。

train_lstm(time_step=args.time_step, val=True)
用于训练的函数,val既是否验证,默认开启。其数据来自get_train_data()

fining_tune_train(time_step=args.time_step)
用于微调模型,如新增数据在原模型继续训练,或者迁移学习等。其数据可以来自get_update_data()

test(time_step=args.time_step)
用于测量测试集的准确率和F1,数据来自get_test_data()

predict(time_step=args.time_step)
用于预测第二天的收盘价,数据来自get_predict_data(args.predict_dir)

5、stock_main.py 主函数

可以调用上述所有函数接口,实现相关功能。

相关论文

《基于LSTM的股票价格的多分类预测》

论文地址:https://www.hanspub.org/journal/PaperInformation.aspx?paperID=32542

About

该项目用于对沪深300股票的预测,包括股票下载,数据清洗,LSTM 模型的训练,测试,以及实时预测


Languages

Language:Python 100.0%