Edit model card

Danish BERT for emotion classification

The BERT Emotion model classifies a Danish text in one of the following class:

  • Glæde/Sindsro
  • Tillid/Accept
  • Forventning/Interrese
  • Overasket/Målløs
  • Vrede/Irritation
  • Foragt/Modvilje
  • Sorg/trist
  • Frygt/Bekymret

It is based on the pretrained Danish BERT model by BotXO which has been fine-tuned on social media data.

This model should be used after detecting whether the text contains emotion or not, using the binary BERT Emotion model.

See the DaNLP documentation for more details.

Here is how to use the model:

from transformers import BertTokenizer, BertForSequenceClassification

model = BertForSequenceClassification.from_pretrained("alexandrainst/da-emotion-classification-base")
tokenizer = BertTokenizer.from_pretrained("alexandrainst/da-emotion-classification-base")

Training data

The data used for training has not been made publicly available. It consists of social media data manually annotated in collaboration with Danmarks Radio.

Downloads last month
140
Inference Examples
This model does not have enough activity to be deployed to Inference API (serverless) yet. Increase its social visibility and check back later, or deploy to Inference Endpoints (dedicated) instead.

Spaces using alexandrainst/da-emotion-classification-base 2