Overview

The stock price prediction program, named for the volatility of Tesla (TSLA) stock caused by Elon Musks' tweets on X, formerly known as Twitter, is a project that bridges the gap between coding and my stock portfolio. This project is still in the early stages of development, as such, it still has a long way to go before it will be deployed for real trading decisions.

(GitHub Link)


Quick Summary

  • Currently, only predictions based on past closing prices, opening prices, high, low, and adjusted close per day are supported.

  • This project is still under construction.

  • Future prospects:

    • Model training will be improved by implementing data regarding bollinger bands, moving averages, and stochastic oscillators.
    • Implementing a web scraper combined with the Hugging face transformer library and a localized and distilled LLM to quantify the effects of Musk’s tweets on the stocks being tracked.

Key Features

  • Applied a Long Short-Term Memory (LSTM) model leveraging Adaptive Moment Estimation (ADAM) optmizer for gradient descent calculation.
  • As of now, the model is mainly trained on the daily closing prices, opening prices, high, low, and adjusted close of each stock it predicts.
  • In the future, it will take into account opening prices, trade volume, bollinger bands, and the stochastic oscillator.
  • The current dataset used has 6 features and 28 million entries.

Tools Used

  • Cuda
  • cuDNN
  • Adam Optimizer
  • LSTM
  • Jupyter Notebook
  • PyTorch
  • Python

Images

Price Predictions vs Historical Data

Predictions NVDA