语言:

• 中文 许可证: apache-2.0

标签:

• bert
• 自然语言理解
• 情感分析

推理: true

示例:

• 文本: "今天心情不好"

二郎神-MegatronBert-1.3B-情感分析版

• GitHub仓库: 封神榜-LM
• 文档: 封神榜文档

简介

2021年登顶FewCLUE和ZeroCLUE榜单的中文BERT模型，在多个情感分析任务上微调后的版本

模型分类

模型信息

基于二郎神-MegatronBert-13亿，我们在8个中文情感分析数据集（总计227,347条样本）上进行了微调。

下游任务表现

使用方式

from transformers import AutoModelForSequenceClassification
from transformers import BertTokenizer
import torch

tokenizer=BertTokenizer.from_pretrained('IDEA-CCNL/Erlangshen-MegatronBert-1.3B-Sentiment')
model=AutoModelForSequenceClassification.from_pretrained('IDEA-CCNL/Erlangshen-MegatronBert-1.3B-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{封神榜-LM,
  title={封神榜语言模型库},
  author={IDEA-CCNL},
  year={2021},
  howpublished={\url{https://github.com/IDEA-CCNL/Fengshenbang-LM}},
}