dair-ai/emotion
Viewer • Updated • 437k • 18.1k • 457
How to use Islamamro/emotion6-aurora-islamamro with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-classification", model="Islamamro/emotion6-aurora-islamamro") # Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("Islamamro/emotion6-aurora-islamamro")
model = AutoModelForSequenceClassification.from_pretrained("Islamamro/emotion6-aurora-islamamro", device_map="auto")Text classifier fine-tuned from distilbert-base-uncased on dair-ai/emotion (6 classes).
Built, trained and published end-to-end via the Aurora Research Portal by islamamro. Trained on CPU on a 1200-example demo subset.
distilbert-base-uncased (~67M parameters)This is a demonstration of the Aurora build -> train -> publish pipeline on a small subset, not a production model. Model size here is set by the base architecture, which is varied deliberately across the Aurora showcase to span a range of parameter counts. For production, fine-tune on the full dataset.
Base model
distilbert/distilbert-base-uncased