Downloads · 30 days
10
1% of all-time downloads
Amazon_rating_review_model
MahmoudMohamed/Amazon_rating_review_model
Amazon_rating_review_model is a text classification model from MahmoudMohamed. Use it when you need a label for a piece of text. It is set up for transformers.
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.
All-time downloads
1.4K
Public
Parameters
167M
669 MB on disk
Likes
1
Public
Repo makeup
Click a slice to open those files.
.safetensors669 MB · 100%
At a glance
- Task
- Text Classification
- Library
- transformers
- Model type
- bert
- Access
- Public
- Created
- Dec 23, 2023
- Updated
- Jan 8, 2024
- SHA
- 694bd9ce
Try a prompt
Model card
From the Hugging Face model README
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
- Task
- Text Classification
- Library
- transformers
- Type
- bert
- Languages
- en
- Created
- Dec 23, 2023
- Updated
- Jan 8, 2024