Sagemaker Distilbert Emotion

🚀 sagemaker-distilbert-emotion

本模型是 distilbert-base-uncased 在 emotion 數據集上的微調版本。它在評估集上取得了出色的效果，為文本分類任務提供了高精度的解決方案。

🚀 快速開始

本模型是 distilbert-base-uncased 在 emotion 數據集上的微調版本。它在評估集上取得了以下結果：

• 損失值：0.2434
• 準確率：0.9165

📚 詳細文檔

模型信息

訓練過程

訓練超參數

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

• 學習率：3e-05
• 訓練批次大小：32
• 評估批次大小：64
• 隨機種子：42
• 優化器：Adam（β1=0.9，β2=0.999，ε=1e-08）
• 學習率調度器類型：線性
• 學習率調度器熱身步數：500
• 訓練輪數：1
• 混合精度訓練：Native AMP

訓練結果

框架版本

• Transformers 4.12.3
• Pytorch 1.9.1
• Datasets 1.15.1
• Tokenizers 0.10.3

📄 許可證

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