CoolFace
Modelpublic

Zhoyang/yelp-sentiment-roberta-large

sourceHugging Facemitupdated 4mo agoView on Hugging Face
0likes27downloads
Model Card

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

LabelMeaningStars
0Negative1–2 ⭐
1Neutral3 ⭐
2Positive4–5 ⭐

Per-class Performance

ClassPrecisionRecallF1
Negative0.93930.92610.9327
Neutral0.69340.76610.7280
Positive——0.9731

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

VersionKey ChangeWeighted F1Neutral F1
v1BERT-base baseline0.8810.514
v3RoBERTa-base0.8920.573
v4max_len 256→5120.9040.604
v6RoBERTa-large0.9120.612
FinalEDA-driven weights0.93890.647

Citation

Developed as part of Georgia Tech OMSA Practicum, Spring 2026 — PathSynch Labs Pod C.