Reinforcer
Collection
4 items • Updated
How to use reinforcer/distilbert-se-emotions with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-classification", model="reinforcer/distilbert-se-emotions") # Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("reinforcer/distilbert-se-emotions")
model = AutoModelForSequenceClassification.from_pretrained("reinforcer/distilbert-se-emotions", device_map="auto")Base Model: distilbert/distilbert-base-uncased
Model Name: reinforcer/distilbert-se-emotions
Task: Text Classification
Language: English
Training Parameters:
| Metric | Score |
|---|---|
| Accuracy | 91.8% |
| Precision | 92.0% |
| Recall | 91.8% |
| F1 Score | 91.8% |
from transformers import pipeline
classifier = pipeline("text-classification", model="reinforcer/distilbert-se-emotions")
results = classifier("Your text to classify here")
Base model
distilbert/distilbert-base-uncased