qianrenjian / News-Recommendation

An investigation of News Recommendation

Download

There are two branches in this REPO. The backup branch is deprecated and used as backup. You can only clone this main branch by:

git clone -b main --single-branch https://github.com/namespace-Pt/News-Recommendation.git

Instruction

1. Download MIND dataset here
2. Save MIND dataset in a directory, e.g. ~/Data

3. pip install torch==1.9.1+cu111 -f https://download.pytorch.org/whl/torch_stable.html
   

4. Add your email information so that you can get an email every time the model is evaluated:

   cd src
   mkdir data
   echo "email = '[e-mail]'" >> data/email.py
   echo "password = 'your password'" >> data/email.py

5. python -m main.twotower --data-root ~/Data   \
                           --batch-size 8       \
                           --world-size 2       \
                           --news-encoder bert

   • --world-size defines number of gpus in ddp training
   • --news-encoder defines news encoder
   • more parameters can be found in src/utils/manager.py