Distilbert Base Uncased Finetuned Amazon Food Reviews

🚀 Amazon-Food-Reviews-distilBERT-base 情感分析模型

這個模型是基於 DistilBERT 微調的情感分析模型，可用於分析亞馬遜食品評論的情感傾向，具有較高的準確率和性能表現。

🚀 快速開始

模型詳情

• 模型描述：此模型是 distilbert-base-uncased 在 亞馬遜食品評論數據集 上的微調版本。
• 評估集結果：
  • 損失值：0.08
  • 準確率：0.87
  • 精確率：0.71
  • 召回率：0.77
  • F1 值：0.73

用途

直接使用

該模型可用於對亞馬遜食品產品評論進行情感分析。

誤用和超範圍使用

請勿使用該模型故意為他人創造敵對或排斥性的環境。此外，該模型並非用於生成關於人或事件的事實性或真實性內容，因此使用該模型生成此類內容超出了其能力範圍。

風險、侷限性和偏差

通過一些實驗，我們發現該模型可能會產生針對代表性不足群體的有偏差的預測。

我們強烈建議用戶徹底探究其用例的這些方面，以評估該模型的風險。我們建議參考以下偏差評估數據集作為起點：WinoBias、WinoGender、Stereoset。

訓練

訓練數據

作者使用 亞馬遜食品評論數據集 對模型進行訓練。

微調超參數

訓練期間使用了以下超參數：

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

訓練過程

框架版本

• Transformers 4.35.2
• Pytorch 2.1.0+cu118
• Tokenizers 0.15.0

📄 許可證

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