Just Another Emotion Classifier

🚀 感情分類Transformerモデル

このモデルは、入力テキストから感情を分類するために微調整されたTransformerモデルです。入力テキストに対して、事前に選択された感情の単語に対する確率分布を出力します。

🚀 クイックスタート

このモデルは、入力テキストに対する感情分類を行います。具体的には、入力テキストに対して、事前に選択された32種類の感情クラスに対する確率分布を出力します。この32種類の感情クラスは、Empathetic Dialoguesデータセットに含まれる感情クラスの数です。

このモデルは、distilbert-base-uncasedモデルをgo-emotionsデータセットで微調整したものをベースに構築されています。そのモデルには、モデルの学習方法を示すJupyter Notebookファイルも含まれているので、是非確認してみてください。

📚 ドキュメント

学習データ

• このモデルの学習には、go-emotionsとbdotloh/empathetic-dialogues-contextsの2つのデータセットが使用されています。

学習手順

前処理

前処理の詳細は記載されていません。

評価結果

評価結果の詳細は記載されていません。

制限事項とバイアス

EmpatheticDialoguesデータセットに関して、以下の制限事項とバイアスがあります。

1. 回答者が感情ラベルを付けた際に記述したコンテキストの文化的特異性の程度を把握することができません（つまり、p(記述 | 感情, 文化)）。
2. ...