Skip to content

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.

Notifications You must be signed in to change notification settings

cooper-a/review-assist

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

14 Commits
 
 
 
 
 
 
 
 

Repository files navigation

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

About

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.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published