Skip to content

kingsman142/predicting-stock-prices

Repository files navigation

predicting-stock-prices

Using an LSTM to predict stock prices

  1. Download data from https://www.kaggle.com/borismarjanovic/price-volume-data-for-all-us-stocks-etfs/data and place it in a data/ folder in the project root directory such that you have a "data/Data", "data/ETFs", and "data/Stocks" folder
  2. Run python3 train.py
  3. (optional) Run python3 test_ood_stocks.py to test your trained model on OOD (out-of-distribution) stocks, which are stocks your model was not trained on
  4. (optional) Run python3 test_buy_seller.py to simulate a buying and selling bot in an auction house
  5. (optional) Run python3 test_live_data.py to pull live prices off of a stock exchange site and compare the success of the model on a daily basis

TODO:

Possible analysis:

  • Plot effect of simple moving average
  • Plot effect of exponential moving average
  • Show results on predicting of one stock
  • Show results on predicting of multiple stocks
  • Investigate use of L1 vs. MSE loss
  • Write study based around performance of different window sizes (or even hidden state sizes, batch sizes, etc.)
  • Investigate whether there's a difference in fluctuations/spikes between popular stocks and non-popular stocks

About

Using an LSTM to predict stock prices

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages