用于情感分类的微调版 DistilRoBERTa-base 🤬🤢😀😐😭😲

模型说明

DistilRoBERTa-base 是一个用于情感分析的 transformer 模型。我使用《老友记》的剧本对白对该模型进行了微调，目的是从文本数据（特别是网飞剧集或电影的对话）中进行情感分类。该模型可预测 6 种埃克曼情感以及一个中性类别，包括愤怒、厌恶、恐惧、喜悦、中性、悲伤和惊讶。

本模型是 Emotion English DistilRoBERTa-base 和 DistilRoBERTa-base 的微调版本。该模型最初基于 Emotion English DistilRoBERTa-base 的以下数据集进行训练：

随后使用以下数据集进行了微调：

使用方法

from transformers import pipeline
classifier = pipeline("sentiment-analysis", model="michellejieli/emotion_text_classifier")
classifier("I love this!")

输出： [{'label': 'joy', 'score': 0.9887555241584778}]