Zhoyang/yelp-sentiment-roberta-large
027
yelp-sentiment-roberta-large
RoBERTa-large fine-tuned for 3-class sentiment classification on Yelp reviews.
Weighted F1: 0.9389 | Accuracy: 93.76% Evaluated on 40,000 held-out samples.
Labels
Per-class Performance
How to use
from transformers import pipeline
classifier = pipeline( "text-classification", model="Zhoyang/yelp-sentiment-roberta-large" )
result = classifier("The food was amazing!")
{'label': 'LABEL_2', 'score': 0.97}
Training Details
- Dataset: Yelp Academic Dataset (499,229 reviews → 239,870 training samples)
- Base model: roberta-large (355M parameters)
- Hardware: NVIDIA A4000 via Paperspace Gradient
- Key decisions:
- max_len=512 (largest single-step F1 gain: +0.012)
- Class-weighted loss: Neg 3.0 / Neu 8.0 / Pos 1.5
- 7 controlled iterations from BERT-base baseline
Iteration History
Citation
Developed as part of Georgia Tech OMSA Practicum, Spring 2026 — PathSynch Labs Pod C.
