mirror of
https://github.com/dupenf/stock-lstm.git
synced 2024-11-25 16:22:36 +08:00
使用lstm模型,训练股票数据,预测未来一天的开盘/收盘/最高/最低价格。
datasets | ||
__init__.py | ||
d0_download.py | ||
d1_showCand.py | ||
d2_viewer.py | ||
d3_prepareddata.py | ||
LICENSE | ||
m1_model.py | ||
m2_test.py | ||
m3_train.py | ||
m5_predict.py | ||
README.md | ||
saved_weights.pt |
代码实现了lstm模型训练股价。 输入为10天的开盘/收盘/最高/最低, 每次预测y为一天的四个价格。
可以循环预测多天股价。
#下载股票数据,保存为csv文件 python d0_download.py
#归一化,并且生产loader数据集 python d3_prepareddata.py
#训练 python m3_train.py
#预测10天的股价 python m5_predict.py