Bert Imdb

🚀 BERT-IMDB

bert-large-casedをIMDBデータセットで感情分類に特化して学習させたモデルです。

🚀 クイックスタート

BERT-IMDBは、IMDBデータセットを用いて感情分類タスクに特化して学習されたモデルです。

✨ 主な機能

• IMDBデータセットを用いた感情分類タスクに最適化されたBERTモデルです。
• simpletransformersライブラリを使用して学習が行われています。

🔧 技術詳細

学習設定

モデルは、simpletransformersライブラリを使用して、IMDBデータセットの80%を使って感情分類タスクで3エポック学習されました。学習率は1e-5で、ライブラリが学習率スケジュールを使用しています。

結果

モデルは検証セットで90%の分類精度を達成しました。

📄 参考資料

完全な実験内容はtlrリポジトリで確認できます。