Modelpublic

atharvadhumal/bert-amazon-polarity

sourceHugging Faceapache-2.0updated 6mo agoView on Hugging Face

0likes10downloads

Model Card

BERT Fine-tuned on Amazon Polarity

Fine-tuned bert-base-uncased for binary sentiment classification on the Amazon Polarity dataset (3.6M training reviews).

Results

Metric	Score
F1 (macro)	97.06%
Accuracy	97.06%
Test samples	400,000

Usage

python

from transformers import pipeline

classifier = pipeline('sentiment-analysis', model='atharvadhumal/bert-amazon-polarity')
classifier('This product is absolutely amazing!')
# [{'label': 'LABEL_1', 'score': 0.998}]

Training Details

Hyperparameter	Value
Base model	bert-base-uncased
Epochs	3
Batch size	64
Learning rate	2e-5
Precision	fp16
Hardware	NVIDIA V100 / H200 (Northeastern HPC)

See the GitHub repo for full training code.