language:

• zh license: apache-2.0

tags:

• roberta
• 自然语言理解
• 情感分析
• 中文

inference: true

widget:

• text: "今天心情不好"

二郎神-RoBERTa-110M-情感分析版

• 主页:封神榜
• GitHub仓库: 封神榜大模型

简介

基于中文RoBERTa-wwm-ext-base模型，在多个情感分析数据集上微调后的版本

模型分类

模型信息

基于hfl/chinese-roberta-wwm-ext预训练模型，我们在8个中文情感分析数据集（总计227,347条样本）上进行微调得到本情感分析专用版本。

下游任务表现

使用方式

from transformers import BertForSequenceClassification
from transformers import BertTokenizer
import torch

tokenizer=BertTokenizer.from_pretrained('IDEA-CCNL/Erlangshen-Roberta-110M-Sentiment')
model=BertForSequenceClassification.from_pretrained('IDEA-CCNL/Erlangshen-Roberta-110M-Sentiment')

text='今天心情不好'

output=model(torch.tensor([tokenizer.encode(text)]))
print(torch.nn.functional.softmax(output.logits,dim=-1))

引用文献

如果您在研究中使用了本模型，请引用我们的论文：

@article{fengshenbang,
  author    = {张嘉旭等},
  title     = {封神榜1.0：中文认知智能基础体系},
  journal   = {CoRR},
  volume    = {abs/2209.02970},
  year      = {2022}
}

也可引用我们的项目主页：

@misc{封神榜大模型,
  title={封神榜大模型},
  author={IDEA-CCNL},
  year={2021},
  howpublished={\url{https://github.com/IDEA-CCNL/Fengshenbang-LM}},
}