Distilbert Imdb

🚀 distilbert-imdb

該模型是基於IMDB數據集對distilbert-base-uncased進行微調後的版本，能有效進行文本分類任務，在評估集上有較高的準確率。

🚀 快速開始

此模型是 distilbert-base-uncased 在 imdb 數據集上的微調版本（訓練筆記本見 此處）。它在評估集上取得了以下結果：

• 損失值：0.1903
• 準確率：0.928

📚 詳細文檔

模型信息

訓練過程

訓練超參數

訓練過程中使用了以下超參數：

• 學習率：5e - 05
• 訓練批次大小：16
• 評估批次大小：16
• 隨機種子：42
• 優化器：Adam（β1 = 0.9，β2 = 0.999，ε = 1e - 08）
• 學習率調度器類型：線性
• 訓練輪數：1

訓練結果

框架版本

• Transformers 4.15.0
• Pytorch 1.10.0+cu111
• Datasets 1.17.0
• Tokenizers 0.10.3

📄 許可證

本模型採用 Apache - 2.0 許可證。