This project includes training and predicting processes with LSTM for stock data. The characteristics is as fellow:
- Concise and modular
- Support three mainstream deep learning frameworks of pytorch, keras and tensorflow
- Parameters, models and frameworks can be highly customized and modified
- Supports incremental training
- Support predicting multiple indicators at the same time
- Support predicting any number of days
- Support train visualization and log record
Chinese introduction can refer to : https://blog.csdn.net/songyunli1111/article/details/78513811
- 更简洁更模块化
- 支持pytorch,keras和tensorflow三个主流深度学习框架
- 参数、模型、框架可高度定制化修改
- 支持增量式训练
- 支持同时预测多个指标，预测未来任意天数
- 支持训练过程的loss可视化和日志记录
The simultaneous predict results for stock high and low price with pytorch show as follow:
