/stock_predict_with_LSTM

Predict stock with LSTM supporting pytorch, keras and tensorflow

Primary LanguagePythonApache License 2.0Apache-2.0

Predict stock with LSTM

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

The simultaneous predict results for stock high and low price with pytorch show as follow:

predict_high_with_pytorch

predict_low_with_pytorch