Distilroberta Base Finetuned Suicide Depression

🚀 微調自殺抑鬱檢測模型distilroberta-base

本模型是基於distilroberta-base在未知數據集上微調得到的版本，可用於檢測推文中的自殺（標籤1）或抑鬱（標籤0）情況。該模型在評估集上取得了如下結果：

• 損失值：0.6622
• 準確率：0.7158

🚀 快速開始

本模型是在SDCNL數據集上微調的Transformer模型的概念驗證（POC），用於檢測推文中的自殺（標籤1）或抑鬱（標籤0）情況。請勿將其用於生產環境。

✨ 主要特性

• 基於distilroberta-base模型微調，能夠對推文中的自殺或抑鬱傾向進行檢測。
• 在評估集上有一定的準確率表現。

📚 詳細文檔

訓練和評估數據

相關信息待補充。

訓練過程

訓練超參數

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

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

訓練結果

框架版本

• Transformers 4.11.3
• Pytorch 1.9.0+cu111
• Datasets 1.13.0
• Tokenizers 0.10.3

📄 許可證

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

⚠️ 重要提示

本模型僅為概念驗證（POC），請勿將其用於生產環境。

💡 使用建議

由於訓練和評估數據的詳細信息暫未提供，在使用該模型時需謹慎評估其適用性和可靠性。