metadata
language:
- en
metrics:
- accuracy
pipeline_tag: text-classification
BERT-Based Amazon Reviews Rating Prediction
Overview
This repository contains a BERT-based model trained on a huge dataset of 500K Amazon reviews to predict ratings on a scale of 1 to 5.
- Model Architecture: BERT Base
- Task: Amazon Reviews Rating Prediction
- Rating Scale: 1 to 5