mirror of
https://github.com/twitter/the-algorithm-ml.git
synced 2024-11-16 13:19:23 +01:00
Source code for Twitter's Recommendation Algorithm
cc39b92c3c
My service to twitter 😇😀 |
||
---|---|---|
.github/workflows | ||
common | ||
core | ||
images | ||
machines | ||
metrics | ||
ml_logging | ||
optimizers | ||
projects | ||
reader | ||
tools | ||
.gitignore | ||
.pre-commit-config.yaml | ||
COPYING | ||
LICENSE.torchrec | ||
model.py | ||
pyproject.toml | ||
README.md |
This project open sources some of the ML models used at Twitter.
Currently these are:
-
The "For You" Heavy Ranker (projects/home/recap).
-
TwHIN embeddings (projects/twhin) https://arxiv.org/abs/2202.05387
This project can be run inside a python virtualenv. We have only tried this on Linux machines and because we use torchrec it works best with an Nvidia GPU. To setup run
./images/init_venv.sh
(Linux only).
The READMEs of each project contain instructions about how to run each project.