My-Machine-Learning-Projects-CT / the-algorithm-ml

Source code for Twitter's Recommendation Algorithm

Home Page:https://blog.twitter.com/engineering/en_us/topics/open-source/2023/twitter-recommendation-algorithm

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

This project open sources some of the ML models used at Twitter.

Currently these are:

  1. The "For You" Heavy Ranker (projects/home/recap).

  2. 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.

About

Source code for Twitter's Recommendation Algorithm

https://blog.twitter.com/engineering/en_us/topics/open-source/2023/twitter-recommendation-algorithm

License:GNU Affero General Public License v3.0


Languages

Language:Python 99.3%Language:Shell 0.7%