#41 · Primary category: Time Series Machine Learning
stock-rnn
Predict stock market prices using RNN model with multilayer LSTM cells + optional multi-stock embeddings.
Project last updated:07/28/22
GitHub Stars
2.0K
Forks
673
Contributors
3
License
Other
Why we included this project
Lilian Weng built this as a teaching project rather than a forecasting tool, and it shows its age in the best way: the code is plain enough to read the architecture directly, and the two companion blog posts explain the reasoning behind each piece. The multi-stock setup is the interesting part. Instead of training one model per ticker, you feed embedding vectors for each stock into the network alongside the price series, so a single LSTM learns patterns across many tickers at once. The fetch and train scripts run without much ceremony, and the README walks you through pulling S&P 500 data from Yahoo Finance and starting TensorBoard. If you want to see how multilayer LSTM cells actually fit together in TensorFlow, this is a solid reference to work from.
Articles for this project
No articles for this project yet.
To suggest a topic or contribute an article, contact us.
Related projects in this category
netdata
The fastest path to AI-powered full stack observability, even for lean teams.
catboost
A fast, scalable, high performance Gradient Boosting on Decision Trees library, used for ranking, classification, regression and other machine learning tasks for Python, R, Java, C++. Supports computation on CPU and GPU.
sktime
A unified framework for machine learning with time series
darts
A python library for user-friendly forecasting and anomaly detection on time series.
featuretools
An open source python library for automated feature engineering