News-Recommendation

News-Recommendation copied to clipboard

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 = '[email protected]'" >> 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