模型:

alexandrainst/da-emotion-classification-base

英文

Danish BERT情绪分类模型

BERT情绪模型将丹麦文本分类为以下类别之一:

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

该模型基于BotXO的预训练 Danish BERT 模型,并在社交媒体数据上进行了微调。

在使用该模型之前,应使用二进制 BERT Emotion model 模型检测文本是否包含情感。

有关更多详细信息,请参见 DaNLP documentation

下面是如何使用该模型的说明:

from transformers import BertTokenizer, BertForSequenceClassification

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

训练数据

用于训练的数据未公开。它包括与丹麦广播公司合作手动注释的社交媒体数据。