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.