/DSML_Project_Reco

Primary LanguageJupyter NotebookMIT LicenseMIT

UCL DSML Project (Thesis)

To pre-train RecoBERT, kindly type "python train.py" in your terminal to train. Hyperparameters can be changed in train.py file.

Fine-tuning RecoBERT is available in two notebooks, Gender fine-tuning, and Age group fine-tuning.