模型:
alexandrainst/da-emotion-classification-base
BERT情绪模型将丹麦文本分类为以下类别之一:
该模型基于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")
用于训练的数据未公开。它包括与丹麦广播公司合作手动注释的社交媒体数据。