Erlangshen MegatronBert 1.3B Sentiment

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

このモデルは、2021年にFewCLUEとZeroCLUEベンチマークでトップを記録した中国語BERTを、いくつかの感情分析タスクでファインチューニングしたバージョンです。

🚀 クイックスタート

• Github: Fengshenbang-LM
• Docs: Fengshenbang-Docs

✨ 主な機能

このモデルは、中国語の感情分析に特化しており、高い精度で感情を判定することができます。

📚 ドキュメント

🔍 モデル分類

🔍 モデル情報

Erlangshen-MegatronBert-1.3Bをベースに、収集した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))

📄 ライセンス

このモデルはApache-2.0ライセンスの下で提供されています。

🔖 引用

もしあなたの研究や開発でこのモデルを使用した場合、以下の文献を引用してください。

論文引用

@article{fengshenbang,
  author    = {Jiaxing Zhang and Ruyi Gan and Junjie Wang and Yuxiang Zhang and Lin Zhang and Ping Yang and Xinyu Gao and Ziwei Wu and Xiaoqun Dong and Junqing He and Jianheng Zhuo and Qi Yang and Yongfeng Huang and Xiayu Li and Yanghan Wu and Junyu Lu and Xinyu Zhu and Weifeng Chen and Ting Han and Kunhao Pan and Rui Wang and Hao Wang and Xiaojun Wu and Zhongshen Zeng and Chongpei Chen},
  title     = {Fengshenbang 1.0: Being the Foundation of Chinese Cognitive Intelligence},
  journal   = {CoRR},
  volume    = {abs/2209.02970},
  year      = {2022}
}

ウェブサイト引用

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