Skip to content

The BERT Product Rating Predictor is a natural language processing model based on the Bidirectional Encoder Representations from Transformers (BERT) model developed to predict star ratings for textual product reviews. 2020.

Notifications You must be signed in to change notification settings

csbanon/bert-product-rating-predictor

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

80 Commits
 
 
 
 
 
 
 
 

Repository files navigation

BERT Product Rating Predictor

Overview

  • Year: 2020
  • Language(s): Python, R
  • Discipline(s): Machine Learning, Natural Language Processing (NLP)
  • Keywords: Amazon Reviews, BERT, Classification, Clustering, Machine Learning, NLP

Description

The BERT Product Rating Predictor is a natural language processing model based on the Bidirectional Encoder Representations from Transformers (BERT) model developed to predict star ratings for textual product reviews. It was created by fine-tuning the BERT model, training it with a custom dataset containing 195,765 reviews gathered from Amazon’s electronic products section. From this model, a k-means clustering model was then employed to explore interesting relationships between the predicted star ratings and their associated reviews.

This model was then used as part of a research project titled Using the BERT Model to Predict and Analyze Star Ratings from Reviews in Amazon Electronic Products.

Team Members

  • Alexander Roustai
  • Jin Koay
  • David Wecke
  • Ankita Tripathi
  • Carlos Santiago Bañón

Build Instructions

  1. Download the bert-product-rating-predictor.ipynb Jupyter notebook.
  2. Download the pretrained model.
  3. In the notebook, set up the pretrained model by adding the pretrained model to the same directory and including it as the file_path. More instructions can be found in the notebook.
  4. Reach the end of the notebook and define a custom review.
  5. Run the notebook. The last block will contain your predicted star rating.

About

The BERT Product Rating Predictor is a natural language processing model based on the Bidirectional Encoder Representations from Transformers (BERT) model developed to predict star ratings for textual product reviews. 2020.

Topics

Resources

Stars

Watchers

Forks