edge555/Review-Rater

A model for rating reviews with stars and confidence using FastText.

Review Rater

Review rating model to rate reviews with stars and confidence using FastText.

Requirements

• FastText

Dataset

• yelp_academic_dataset_review.json

How to run

• Install FastText in the directory.

• Download the json file.

• Run split_data.py to produce training and testing datasets.

• Run train_model.py to train and save the model.

• Run main.py to run the model. Change the 'reviews' array to test different reviews.