nyu-mll/glue
Viewer • Updated • 1.49M • 430k • 523
A 3-class sentiment classifier (Negative / Neutral / Positive) fine-tuned from
google/muril-base-cased, which unlike the
v1 roberta-base model can natively handle Hindi (Devanagari) and Hinglish (romanized
code-mixed Hindi-English) text in addition to English.
| Base model | google/muril-base-cased |
| Task | Text classification (3-class sentiment) |
| Languages | English, Hindi, Hinglish (code-mixed) |
| Training rows | 96,761 (train) / 8,538 (val) / 8,538 (test) |
| Max sequence len | 128 tokens |
Classes were capped per-label and a class-weighted loss was used during training to reduce the effect of English data outnumbering Hindi/Hinglish data.
precision recall f1-score support
Negative 0.90 0.85 0.87 3000
Neutral 0.81 0.89 0.85 2538
Positive 0.87 0.84 0.86 3000
accuracy 0.86 8538
macro avg 0.86 0.86 0.86 8538
weighted avg 0.86 0.86 0.86 8538
from transformers import pipeline
clf = pipeline("text-classification", model="airzipm/sentiment-analysis-muril-v2")
print(clf("ye movie achi hai"))
print(clf("यह फिल्म बहुत अच्छी है"))
print(clf("This was a great experience!"))