How to use Sathwik3/distilbert-emotion-classifier with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="Sathwik3/distilbert-emotion-classifier")
# Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("Sathwik3/distilbert-emotion-classifier") model = AutoModelForSequenceClassification.from_pretrained("Sathwik3/distilbert-emotion-classifier")
The accuracy and weighted F1 scores are mentioned for this model, can you also mention the accuracy for the model for each class?Also, what dataset is used for this emotion classification?
· Sign up or log in to comment
