使用lstm模型,训练股票数据,预测未来一天的开盘/收盘/最高/最低价格。
Go to file
2024-08-04 19:08:45 +08:00
datasets v1.0 2024-08-04 19:05:34 +08:00
__init__.py v1.0 2024-08-04 19:05:34 +08:00
d0_download.py v1.0 2024-08-04 19:05:34 +08:00
d1_showCand.py v1.0 2024-08-04 19:05:34 +08:00
d2_viewer.py v1.0 2024-08-04 19:05:34 +08:00
d3_prepareddata.py v1.0 2024-08-04 19:05:34 +08:00
LICENSE Initial commit 2024-08-04 19:03:55 +08:00
m1_model.py v1.0 2024-08-04 19:05:34 +08:00
m2_test.py v1.0 2024-08-04 19:05:34 +08:00
m3_train.py v1.0 2024-08-04 19:05:34 +08:00
m5_predict.py v1.0 2024-08-04 19:05:34 +08:00
README.md Update README.md 2024-08-04 19:08:45 +08:00
saved_weights.pt v1.0 2024-08-04 19:05:34 +08:00

代码实现了lstm模型训练股价。 输入为10天的开盘/收盘/最高/最低, 每次预测y为一天的四个价格。

可以循环预测多天股价。

#下载股票数据保存为csv文件 python d0_download.py

#归一化并且生产loader数据集 python d3_prepareddata.py

#训练 python m3_train.py

#预测10天的股价 python m5_predict.py