Emotions Babel
Collection
Emotion classification using 6-label and 9-label codebooks.
•
12 items
•
Updated
This repository is publicly accessible, but you have to accept the conditions to access its files and content.
Our models are intended for academic use only. If you are not affiliated with an academic institution, please provide a rationale for using our models. Please allow us a few business days to manually review subscriptions.
Log in or Sign Up to review the conditions and access this model content.
This model is designed for emotion classification in Slovak texts.
It was fine-tuned to recognize nine emotion categories and trained on a dataset with labeled examples.
Label | Emotion |
---|---|
0 | Anger |
1 | Fear |
2 | Disgust |
3 | Sadness |
4 | Joy |
5 | None of them |
6 | Enthusiasm |
7 | Hope |
8 | Pride |
The model was evaluated using precision, recall, f1-score, and accuracy.
Label | Precision | Recall | F1-score | Support |
---|---|---|---|---|
Anger (0) | 0.54 | 0.56 | 0.55 | 777 |
Fear (1) | 0.84 | 0.77 | 0.80 | 776 |
Disgust (2) | 0.92 | 0.95 | 0.93 | 776 |
Sadness (3) | 0.84 | 0.84 | 0.84 | 775 |
Joy (4) | 0.83 | 0.79 | 0.81 | 736 |
None of them (5) | 0.67 | 0.65 | 0.66 | 1594 |
Enthusiasm (6) | 0.61 | 0.64 | 0.63 | 776 |
Hope (7) | 0.51 | 0.54 | 0.53 | 777 |
Pride (8) | 0.78 | 0.80 | 0.79 | 776 |
To use this model for text classification in Python:
from transformers import pipeline
classifier = pipeline("text-classification", model="poltextlab/proposal_2b_german_to_slovak_PT_label_v2")
text = "Ich bin sehr glücklich, dass du hier bist!" # German
result = classifier(text)
print(result)