itserphan/yelp-polarity-roberta

🍔 Yelp Polarity Classifier (RoBERTa-base)

This model is a fine-tuned version of roberta-base on the Yelp Polarity dataset.
It classifies reviews as Positive ⭐ or Negative 👎 with high accuracy.

---

🚀 Quick Use

from transformers import pipeline

classifier = pipeline("text-classification", model="itserphan/yelp-polarity-roberta")

print(classifier("The burger was amazing, I'll definitely come back!"))
# [{'label': 'Positive', 'score': 0.998}]

print(classifier("Terrible service. Food was cold and overpriced."))
# [{'label': 'Negative', 'score': 0.996}]