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
