alexandrainst/da-emotion-classification-base

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.