Distilbert Base Uncased Finetuned Emotion

🚀 distilbert-base-uncased-finetuned-emotion

本模型是 distilbert-base-uncased 在 emotion 數據集上的微調版本。它在文本分類任務中表現出色，能有效對文本情感進行分類，為情感分析等應用提供了強大支持。

📚 詳細文檔

模型評估結果

該模型在評估集上取得了以下成績：

• 損失值（Loss）：0.2175
• 準確率（Accuracy）：0.923
• F1 值（F1）：0.9233

訓練過程

訓練超參數

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

• 學習率（learning_rate）：2e-05
• 訓練批次大小（train_batch_size）：64
• 評估批次大小（eval_batch_size）：64
• 隨機種子（seed）：42
• 優化器（optimizer）：Adam，其中 betas=(0.9, 0.999)，epsilon=1e-08
• 學習率調度器類型（lr_scheduler_type）：線性
• 訓練輪數（num_epochs）：2

訓練結果

框架版本

• Transformers：4.11.3
• Pytorch：1.10.0+cu111
• Datasets：1.16.1
• Tokenizers：0.10.3

📄 許可證

本項目採用 Apache-2.0 許可證。

🔍 模型索引