/review-assist

A fine-tuned bert based model to predict restaurant review stars based on user generated text. Trained using the Yelp Dataset achieving 0.71 F1 Score.

Primary LanguageJupyter Notebook

Review Assist

Review Assist is an assistive tool that helps people write restaurant reviews by predicting the rating (number of stars) of the review being written in real time. Currently the project is in the modeling phase. A fine-tuned BERT model is being trained for the star predictions task.

Contributing

Pull requests are welcome. For major changes, please open an issue first to discuss what you would like to change.

License

MIT