Edit model card

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
Safetensors
Model size
117M params
Tensor type
F32
·

Dataset used to train botdevringring/nl-naxai-ai-emotion-classification-101608122023

Evaluation results