语言: 英语
标签:
- 文本分类
- PyTorch
- RoBERTa
- 情感分析
数据集:
- go_emotions
许可证: MIT
小部件示例:
该模型针对GoEmotions数据集训练，该数据集包含标注了28种情感的58k条Reddit评论
- 情感类别包括: 钦佩、有趣、愤怒、烦恼、赞同、关心、困惑、好奇、渴望、失望、反对、厌恶、尴尬、兴奋、恐惧、感激、悲伤、快乐、爱、紧张、乐观、自豪、领悟、宽慰、悔恨、伤心、惊讶 + 中性
训练细节:
- 训练脚本在此处提供: https://github.com/bsinghpratap/roberta_train_goEmotion
- 如果您在运行模型时遇到问题，请在仓库中提交issue，我会尽快回复。
- 该模型在大多数情感上表现良好，除了: '渴望'、'厌恶'、'尴尬'、'兴奋'、'恐惧'、'悲伤'、'紧张'、'自豪'、'宽慰'、'悔恨'、'惊讶'
- 我将尝试进一步微调模型，如果RoBERTa取得更好的性能，会在此更新。
- 每个文本数据点可能有多个标签。大多数训练集数据只有一个标签: Counter({1: 36308, 2: 6541, 3: 532, 4: 28, 5: 1})。因此目前我只使用了每个数据点的第一个标签。虽然不理想，但效果尚可。
模型性能
============================================================
情感: 钦佩
============================================================
GoEmotions论文: 0.65
RoBERTa: 0.62
支持数: 504
============================================================
情感: 有趣
============================================================
GoEmotions论文: 0.80
RoBERTa: 0.78
支持数: 252
============================================================
情感: 愤怒
============================================================
GoEmotions论文: 0.47
RoBERTa: 0.44
支持数: 197
============================================================
情感: 烦恼
============================================================
GoEmotions论文: 0.34
RoBERTa: 0.22
支持数: 286
============================================================
情感: 赞同
============================================================
GoEmotions论文: 0.36
RoBERTa: 0.31
支持数: 318
============================================================
情感: 关心
============================================================
GoEmotions论文: 0.39
RoBERTa: 0.24
支持数: 114
============================================================
情感: 困惑
============================================================
GoEmotions论文: 0.37
RoBERTa: 0.29
支持数: 139
============================================================
情感: 好奇
============================================================
GoEmotions论文: 0.54
RoBERTa: 0.48
支持数: 233
============================================================
情感: 失望
============================================================
GoEmotions论文: 0.28
RoBERTa: 0.18
支持数: 127
============================================================
情感: 反对
============================================================
GoEmotions论文: 0.39
RoBERTa: 0.26
支持数: 220
============================================================
情感: 感激
============================================================
GoEmotions论文: 0.86
RoBERTa: 0.84
支持数: 288
============================================================
情感: 快乐
============================================================
GoEmotions论文: 0.51
RoBERTa: 0.47
支持数: 116
============================================================
情感: 爱
============================================================
GoEmotions论文: 0.78
RoBERTa: 0.68
支持数: 169
============================================================
情感: 中性
============================================================
GoEmotions论文: 0.68
RoBERTa: 0.61
支持数: 1606
============================================================
情感: 乐观
============================================================
GoEmotions论文: 0.51
RoBERTa: 0.52
支持数: 120
============================================================
情感: 领悟
============================================================
GoEmotions论文: 0.21
RoBERTa: 0.15
支持数: 109
============================================================
情感: 伤心
============================================================
GoEmotions论文: 0.49
RoBERTa: 0.42
支持数: 108
Transformers%20--%3e%3cdefs%3e%3cstyle%3e%20.st0%20{%20fill:%20none;%20}%20.st1%20{%20fill:%20%235ce6cf;%20}%20.st2%20{%20fill:%20%230080ff;%20}%20.st3%20{%20clip-path:%20url(%23clippath);%20}%20%3c/style%3e%3cclipPath%20id='clippath'%3e%3crect%20class='st0'%20y='0'%20width='120'%20height='32'/%3e%3c/clipPath%3e%3c/defs%3e%3cg%20class='st3'%3e%3cg%3e%3cpath%20class='st1'%20d='M22.4,32H3.2c-1,0-2-.5-2.6-1.4-.6-.9-.7-2-.4-2.9L9.8,2.1C10.3.8,11.5,0,12.8,0s2.5.8,3,2.1l9.6,25.6c.4,1,.2,2.1-.4,2.9-.6.9-1.6,1.4-2.6,1.4ZM7.8,25.6h10l-5-13.3-5,13.3Z'/%3e%3cpath%20class='st2'%20d='M33.6,32c-1.3,0-2.5-.8-3-2.1L21,4.3c-.6-1.7.2-3.5,1.9-4.1,1.7-.6,3.5.2,4.1,1.9l9.6,25.6c.6,1.7-.2,3.5-1.9,4.1-.4.1-.8.2-1.1.2h0Z'/%3e%3cpath%20d='M71.8,9.1h-6.9v4.6h6.9c2.5,0,4.6,2,4.6,4.6h-6.9c-3.8,0-6.9,3.1-6.9,6.9s3.1,6.9,6.9,6.9h2.3c1.7,0,3.2-.4,4.6-1.2v1.2h4.6v-13.7c0-5-4.1-9.1-9.1-9.1h0ZM71.8,27.4h-2.3c-1.3,0-2.3-1-2.3-2.3s1-2.3,2.3-2.3h6.9c0,2.5-2,4.6-4.6,4.6h0Z'/%3e%3cpath%20d='M52.3,9.1h-4.6V0h-4.6V32h9.1c5,0,9.1-4.1,9.1-9.1v-4.6c0-5.1-4.1-9.1-9.1-9.1h0ZM56.9,22.9c0,2.5-2,4.6-4.6,4.6h-4.6v-13.7h4.6c2.5,0,4.6,2,4.6,4.6v4.6h0Z'/%3e%3cpath%20d='M110.9,13.7h9.1v-4.6h-9.1c-5.1,0-9.1,4.1-9.1,9.1v4.6c0,5.1,4.1,9.1,9.1,9.1h9.1v-4.6h-9.1c-2.5,0-4.6-2-4.6-4.6h13.7v-4.6h-13.7c0-2.5,2-4.6,4.6-4.6h0Z'/%3e%3cpath%20d='M93.6,18.3h-4.6c-1.3,0-2.3-1-2.3-2.3s1-2.3,2.3-2.3h10.3v-4.6h-10.3c-3.8,0-6.9,3.1-6.9,6.9s3.1,6.9,6.9,6.9h4.6c1.3,0,2.3,1,2.3,2.3s-1,2.3-2.3,2.3h-10.3v4.6h10.3c3.8,0,6.9-3.1,6.9-6.9s-3.1-6.9-6.9-6.9Z'/%3e%3c/g%3e%3c/g%3e%3c/svg%3e)