ruBert-base-russian-emotion-detectionオープンソースモデル - ロシア語の9種類の感情分類を高精度で行う

ホーム

Rubert Base Russian Emotion Detection

MaxKazakによって開発

ruBert-baseをファインチューニングしたロシア語マルチラベル感情分類モデルで、ru_goemotionsデータセットで訓練され、9種類の感情分類をサポート

テキスト分類

Transformers

その他オープンソースライセンス:Apache-2.0 #ロシア語感情分析 #マルチラベル分類 #高AUC性能

ダウンロード数 1,288

リリース時間 : 5/28/2023

モデル概要

このモデルはロシア語テキストの感情分析専用で、喜び、興味、驚き、悲しみ、怒り、嫌悪、恐怖、罪悪感、中立感情など、テキストに表現された複数の感情を識別できます。

モデル特徴

マルチラベル感情分類

テキストに表現された複数の感情を同時に識別可能で、単一感情に限定されない

高精度

評価データセットで92.4%のAUC値を達成し、優れた分類性能を発揮

幅広い感情カバレッジ

9種類の異なる感情分類をサポートし、広範な人間の感情表現をカバー

モデル能力

ロシア語テキスト感情分析

マルチラベル分類

感情認識

使用事例

ソーシャルメディア分析

ユーザーコメント感情分析

ソーシャルメディア上のロシア語ユーザーコメントの感情を分析

複数の複合感情を正確に識別

カスタマーサービス

顧客フィードバック感情分類

ロシア語の顧客フィードバックにおける感情傾向を自動分類

ネガティブな感情フィードバックの迅速な識別を支援

🚀 ruBert-base-russian-emotions-classifier-goEmotions

このモデルは、ai-forever/ruBert-base を Djacon/ru_goemotions でファインチューニングしたバージョンです。評価セット（2エポック目）では、以下の結果を達成しています。

損失: 0.2088
AUC: 0.9240

テストデータセットにおける予測確率の品質は以下の通りです。

ラベル	喜び	関心	驚き	悲しみ	怒り	嫌悪	恐怖	罪悪感	中立	平均
AUC	0.9369	0.9213	0.9325	0.8791	0.8374	0.9041	0.9470	0.9758	0.8518	0.9095
F1-micro	0.9528	0.9157	0.9697	0.9284	0.8690	0.9658	0.9851	0.9875	0.7654	0.9266
F1-macro	0.8369	0.7922	0.7561	0.7392	0.7351	0.7356	0.8176	0.8247	0.7650	0.7781

📚 詳細ドキュメント

トレーニング手順

トレーニングハイパーパラメータ

トレーニング中に使用されたハイパーパラメータは以下の通りです。

learning_rate: 5e-05
train_batch_size: 16
eval_batch_size: 16
seed: 42
optimizer: Adam (betas=(0.9,0.999), epsilon=1e-08)
lr_scheduler_type: linear
num_epochs: 3

トレーニング結果

トレーニング損失	エポック	ステップ	検証損失	AUC
0.1755	1.0	1685	0.1717	0.9220
0.1391	2.0	3370	0.1757	0.9240
0.0899	3.0	5055	0.2088	0.9106

フレームワークバージョン

Transformers 4.24.0
Pytorch 2.0.1
Datasets 2.12.0
Tokenizers 0.11.0

📄 ライセンス

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

🔍 その他の情報

言語

ロシア語

データセット

Djacon/ru_goemotions

評価指標

正解率

ウィジェット

テキスト: Очень рад тебя видеть!
テキスト: Как дела?
テキスト: Мне немного отвратно это делать
テキスト: Я испытал мурашки от страха
テキスト: Нет ничего радостного в этих горьких новостях
テキスト: Ого, неожидал тебя здесь увидеть!
テキスト: Фу ну и мерзость
テキスト: Мне неприятно общение с тобой

ベースモデル

ai-forever/ruBert-base

モデルインデックス

名前: ruBert-base-russian-emotions-classifier-goEmotions 結果:
- タスク: タイプ: マルチラベルテキスト分類名前: マルチラベルテキスト分類データセット: 名前: ru_goemotions タイプ: Djacon/ru_goemotions 引数: ru 評価指標:
  - タイプ: roc_auc 値: 92% 名前: マルチラベルROC AUC