pip install transformers sklearn pandas
To run training and evaluation just do:
cd NLP_CW
python NLP2077.py
The results will be printed and model saved in NLP_CW/models
- We trained our model on an nvidia v100s GPU with 32GB of memory. Please change the batchsize on line 27 or use CPU if you are out of memory
- Yitao Qiu yq1321@ic.ac.uk
- Jonas Brami jb2021@ic.ac.uk
- Zifeng Tang zt520@ic.ac.uk