Cat Emotion Classifier

🚀 vit-base-cat-emotions

このモデルは、猫の感情分類を行うための画像分類モデルです。ユーザーは、こちらでモデルを試すことができ、詳細についてはGitHubリポジトリを参照できます。

🚀 クイックスタート

このモデルは、google/vit-base-patch16-224-in21k をカスタムデータセットでファインチューニングしたバージョンです。評価セットでは以下の結果を達成しています。

• 損失: 1.0160
• 正解率: 0.6353

📚 ドキュメント

モデル情報

モデルの結果

訓練手順

訓練ハイパーパラメータ

訓練時には以下のハイパーパラメータが使用されました。

• 学習率 (learning_rate): 0.0002
• 訓練バッチサイズ (train_batch_size): 16
• 評価バッチサイズ (eval_batch_size): 8
• シード (seed): 42
• オプティマイザー (optimizer): Adam (betas=(0.9,0.999), epsilon=1e-08)
• 学習率スケジューラーの種類 (lr_scheduler_type): linear
• エポック数 (num_epochs): 10
• 混合精度訓練 (mixed_precision_training): Native AMP

訓練結果

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

• Transformers 4.44.1
• Pytorch 2.2.2+cu118
• Datasets 2.20.0
• Tokenizers 0.19.1

📄 ライセンス

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