Yelp Review Rating Prediction: Machine Learning and Deep Learning Models

12/12/2020
by   Zefang Liu, et al.
3

We predict restaurant ratings from Yelp reviews based on Yelp Open Dataset. Data distribution is presented, and one balanced training dataset is built. Two vectorizers are experimented for feature engineering. Four machine learning models including Naive Bayes, Logistic Regression, Random Forest, and Linear Support Vector Machine are implemented. Four transformer-based models containing BERT, DistilBERT, RoBERTa, and XLNet are also applied. Accuracy, weighted F1 score, and confusion matrix are used for model evaluation. XLNet achieves 70 Regression with 64

READ FULL TEXT

page 4

page 6

research
01/19/2023

SpotHitPy: A Study For ML-Based Song Hit Prediction Using Spotify

In this study, we approached the Hit Song Prediction problem, which aims...
research
06/13/2021

SASICM A Multi-Task Benchmark For Subtext Recognition

Subtext is a kind of deep semantics which can be acquired after one or m...
research
03/10/2018

Influence of the Event Rate on Discrimination Abilities of Bankruptcy Prediction Models

In bankruptcy prediction, the proportion of events is very low, which is...
research
12/30/2019

AutoDiscern: Rating the Quality of Online Health Information with Hierarchical Encoder Attention-based Neural Networks

Patients increasingly turn to search engines and online content before, ...
research
07/06/2023

Machine Learning to detect cyber-attacks and discriminating the types of power system disturbances

This research proposes a machine learning-based attack detection model f...
research
09/15/2017

Road Friction Estimation for Connected Vehicles using Supervised Machine Learning

In this paper, the problem of road friction prediction from a fleet of c...

Please sign up or login with your details

Forgot password? Click here to reset