jasonwei20 / yelp_rating_predictor

Predict number of stars based on text review

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

yelp_rating_predictor

Predict number of stars based on text review.

Achieves accuracy of 62 percent on cross validation set.

Gets average word2vec embedding of text review and puts it into a neural net and then softmax.

About

Predict number of stars based on text review


Languages

Language:Jupyter Notebook 95.7%Language:Python 4.3%