Accomodation Room Classification

🚀 宿泊施設の部屋分類モデル

このモデルは、画像分類タスクに特化したモデルです。google/vit-base-patch16-224 を imagefolder データセットでファインチューニングしたもので、評価セットで良好な精度を達成しています。

🚀 クイックスタート

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

• 損失: 0.3910
• 精度: 0.875

📚 ドキュメント

モデルの詳細

このモデルに関する詳細情報は、今後のアップデートで追加予定です。

想定される用途と制限

このモデルの想定用途と制限に関する詳細情報は、今後のアップデートで追加予定です。

学習と評価データ

学習と評価に使用したデータに関する詳細情報は、今後のアップデートで追加予定です。

学習手順

学習ハイパーパラメータ

学習中に使用されたハイパーパラメータは以下の通りです。

• 学習率: 5e-05
• 学習バッチサイズ: 32
• 評価バッチサイズ: 32
• シード: 42
• 勾配累積ステップ: 4
• 総学習バッチサイズ: 128
• オプティマイザ: Adam (betas=(0.9,0.999), epsilon=1e-08)
• 学習率スケジューラの種類: 線形
• 学習率スケジューラのウォームアップ比率: 0.1
• エポック数: 7

学習結果

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

• Transformers 4.40.0
• Pytorch 2.2.1+cu121
• Datasets 2.19.0
• Tokenizers 0.19.1

📄 ライセンス

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

情報テーブル