AndyTKH / DSML_Project_Reco

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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.

About

License:MIT License


Languages

Language:Jupyter Notebook 91.5%Language:Python 8.5%