← all repositories
borisbanushev/stockpredictionai

An everything-bagel of 2019 AI, applied to one stock ticker

An elaborate Jupyter notebook that pipelines nearly every trendy 2019 deep-learning technique into a single end-to-end stock-prediction experiment for Goldman Sachs.

5.6k stars JavaScript Domain AppsML Frameworks
stockpredictionai
Not currently ranked — collecting fresh signals.
star history

What it does

This is a monolithic tutorial notebook, not a library. It builds a complete pipeline to predict daily price movements of Goldman Sachs stock by fusing historical prices, technical indicators, news sentiment, correlated assets, and options data. The centerpiece is a Generative Adversarial Network—an LSTM generator paired with a 1D CNN discriminator—trained in MXNet/Gluon.

The interesting bit

The author treats the project as a kitchen-sink demonstration of circa-2019 machine learning: BERT for NLP sentiment, Fourier transforms for trend decomposition, stacked autoencoders for hidden feature extraction, ARIMA, PCA eigen-portfolios, and self-organized maps for options anomaly detection. Hyperparameters are tuned via both Bayesian optimization with Gaussian processes and reinforcement learning using Rainbow and PPO, which is overkill in a charmingly academic way.

Key highlights

  • GAN architecture with LSTM generator and CNN discriminator for time-series synthesis
  • Hyperparameter search via Bayesian optimization and RL agents (Rainbow, PPO)
  • Feature engineering spans BERT sentiment, Fourier transforms, ARIMA, stacked autoencoders, and PCA
  • Statistical sanity checks (heteroskedasticity, multicollinearity, serial correlation) and XGBoost feature importance
  • Built on MXNet/Gluon and designed for multi-GPU training

Caveats

  • It is a single narrative notebook from January 2019, not a maintained package or reusable framework
  • The repository is tagged as JavaScript, though the work is clearly Python-based (Jupyter/MXNet)
  • The README is long on methodology but truncated before showing concrete results or performance metrics

Verdict

Worth a skim if you want to see how pre-Transformer-hype deep learning pieces fit together in finance; avoid if you need a robust backtesting framework or a library to import.

Frequently asked

What is borisbanushev/stockpredictionai?
An elaborate Jupyter notebook that pipelines nearly every trendy 2019 deep-learning technique into a single end-to-end stock-prediction experiment for Goldman Sachs.
Is stockpredictionai open source?
Yes — borisbanushev/stockpredictionai is an open-source project tracked on heatdrop.
What language is stockpredictionai written in?
borisbanushev/stockpredictionai is primarily written in JavaScript.
How popular is stockpredictionai?
borisbanushev/stockpredictionai has 5.6k stars on GitHub.
Where can I find stockpredictionai?
borisbanushev/stockpredictionai is on GitHub at https://github.com/borisbanushev/stockpredictionai.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.