Dataset
Install the requirements.yaml to your conda env to run all the code in this project.
conda env update -b my_env --file requirements.yaml
To train the Bert model, run python train.py. To test the bert model, update the variable 'text' in test.py and execute python train.py . The model weights are located in the 'results' folder. To train and run the Naive Bayes model, please go into the 'Research.ipynb' file and execute all blocks of code.
Website: https://yelp-recommendation.ue.r.appspot.com/report