Fine-grained Sentiment (SST-5) - Aurora - islamamro
Sentiment on a 5-level scale from very negative to very positive.
Fine-tuned from distilbert-base-uncased on SetFit/sst5. Built, trained and published end-to-end via the Aurora Research Portal by islamamro on an NVIDIA RTX 3090.
- Task: text classification (5 classes)
- Held-out accuracy: 0.45
Use it
from transformers import pipeline
clf = pipeline("text-classification", model="Islamamro/sst5-sentiment-aurora-islamamro")
clf("your text here")
Note
Trained on a 1,400-example demo subset - a proof of the Aurora build->train->publish pipeline, not a production model. Fine-tune on full data for real use.
- Downloads last month
- 27