botdevringring/nl-naxai-ai-emotion-classification-101608122023(latest)
The model is trained on the emotion classification task in Dutch. It uses 6 labels: anger, fear, joy love, sadness, surprise.
This model is finetuned from robbert-v2-dutch-base. RobBERT is the state-of-the-art Dutch BERT model. It is a large pre-trained general Dutch language model that can be fine-tuned on a given dataset to perform any text classification, regression or token-tagging task.
Model Details
- Language: nl
- Problem type: Multi-class Classification
- Model Architecture: RobBERT
- Model Name: nl-naxai-ai-emotion-classification-101608122023
- Creation Date: 10:16h 08/12/2023
Dataset Summary
The botdevringring/NL_emotion_classification dataset is a balanced dataset translated from the English Twitter messages Emotion dataset of with six basic emotions: anger, fear, joy, love, sadness, and surprise. For more detailed information please refer to the paper.
Classification Report:
Label | Precision | Recall | f1-Score | Support |
---|---|---|---|---|
anger | 0.91 | 0.91 | 0.91 | 2956 |
fear | 0.89 | 0.83 | 0.86 | 2956 |
joy | 0.89 | 0.83 | 0.86 | 2956 |
love | 0.88 | 0.91 | 0.89 | 2956 |
sadness | 0.91 | 0.89 | 0.90 | 2956 |
surprise | 0.87 | 0.97 | 0.92 | 2956 |
How to use this model
You can use Python to access this model:
from transformers import pipeline
analyzer = pipeline(
task='text-classification',
model='botdevringring/nl-naxai-ai-emotion-classification-101608122023',
tokenizer='botdevringring/nl-naxai-ai-emotion-classification-101608122023'
)
result = analyzer(
"ik blog omdat het iets is waar ik een passie voor heb."
)
result
[
{
'label': 'love',
'score': 0.9728550314903259
}
]
Or you can use cURL:
curl https://api-inference.huggingface.co/models/botdevringring/nl-naxai-ai-emotion-classification-101608122023 \
-X POST \
-d '{"inputs": "ik blog omdat het iets is waar ik een passie voor heb"}' \
-H 'Content-Type: application/json' \
-H "Authorization: Bearer <Your HF API token>"
Acknowledgements
Model trained by Eduardo Brigham for Naxai powered by The Ring Ring Company
- Downloads last month
- 341
Dataset used to train botdevringring/nl-naxai-ai-emotion-classification-101608122023
Evaluation results
- Accuracy on emotiontest set self-reported0.896
- Precision Macro on emotiontest set self-reported0.897
- Precision Micro on emotiontest set self-reported0.896
- Precision Weighted on emotiontest set self-reported0.897
- Recall Macro on emotiontest set self-reported0.896
- Recall Micro on emotiontest set self-reported0.896
- Recall Weighted on emotiontest set self-reported0.896
- F1 Macro on emotiontest set self-reported0.895
- F1 Micro on emotiontest set self-reported0.895
- F1 Weighted on emotiontest set self-reported0.895