🚀 二郎神-Roberta-330M-情感分析
中国語のRoBERTa-wwm-ext-largeをいくつかの感情分析タスクでファインチューニングしたバージョンです。
🚀 クイックスタート
このモデルは、中国語のRoBERTa-wwm-ext-largeをベースに、いくつかの感情分析データセットでファインチューニングされたものです。以下に使用方法の例を示します。
✨ 主な機能
- 中国語の感情分析タスクに特化したモデルです。
- 複数の感情分析データセットでファインチューニングされており、高い精度を実現しています。
📦 インストール
このモデルを使用するには、transformers
ライブラリが必要です。以下のコマンドでインストールできます。
pip install transformers
💻 使用例
基本的な使用法
from transformers import BertForSequenceClassification
from transformers import BertTokenizer
import torch
tokenizer=BertTokenizer.from_pretrained('IDEA-CCNL/Erlangshen-Roberta-330M-Sentiment')
model=BertForSequenceClassification.from_pretrained('IDEA-CCNL/Erlangshen-Roberta-330M-Sentiment')
text='今天心情不好'
output=model(torch.tensor([tokenizer.encode(text)]))
print(torch.nn.functional.softmax(output.logits,dim=-1))
📚 ドキュメント
モデル分類
属性 |
詳細 |
モデルタイプ |
二郎神-Roberta-330M-情感分析 |
タスク |
自然言語理解(NLU) - 感情分析 |
シリーズ |
二郎神 |
モデル |
Roberta |
パラメータ |
330M |
追加情報 |
中国語 - 感情分析 |
モデル情報
chinese-roberta-wwm-ext-largeをベースに、収集した8つの中国語分野の感情分析データセット(合計227,347サンプル)で感情分析バージョンをファインチューニングしました。
下流性能
モデル |
ASAP-SENT |
ASAP-ASPECT |
ChnSentiCorp |
Erlangshen-Roberta-110M-Sentiment |
97.77 |
97.31 |
96.61 |
Erlangshen-Roberta-330M-Sentiment |
97.9 |
97.51 |
96.66 |
Erlangshen-MegatronBert-1.3B-Sentiment |
98.1 |
97.8 |
97 |
📄 ライセンス
このモデルは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}},
}