Bert Base Uncased Goemotions Ekman Finetuned

🚀 感情分類モデル

このモデルは、テキストの感情分類を行うためのもので、Go Emotionsデータセットを用いて訓練されています。BERTモデルをベースに、特殊トークンの追加や損失関数の最適化などの工夫が施されています。

🚀 クイックスタート

本モデルは、テキストの感情分類タスクに使用できます。以下の特徴を持っています。

✨ 主な機能

1. 大文字小文字を区別しないBERT事前学習モデルに線形出力層を追加した構成。
2. トークナイザーの特殊トークンリストにいくつかの一般的な絵文字とトークンを追加。
3. 訓練時にラベルスムージングを行い、モデルの汎化性能を向上。
4. 重み付き損失とフォーカル損失を使用して、訓練が不十分なケースを改善。

📄 ライセンス

このプロジェクトはMITライセンスの下で公開されています。