• Stars
    star
    730
  • Rank 62,081 (Top 2 %)
  • Language
    Python
  • License
    Apache License 2.0
  • Created about 7 years ago
  • Updated over 1 year ago

Reviews

There are no reviews yet. Be the first to send feedback to the community and the maintainers!

Repository Details

Predict stock with LSTM supporting pytorch, keras and tensorflow

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