Sagemaker Roberta Base Emotion

🚀 基於SageMaker的RoBERTa-base文本情感分類模型

本項目基於Amazon SageMaker和Hugging Face深度學習容器對RoBERTa-base模型進行微調，用於多類別文本分類（情感檢測）任務，能夠有效識別文本中的情感傾向。

🚀 快速開始

本模型是使用Amazon SageMaker和全新的Hugging Face深度學習容器進行微調訓練的。它主要用於解決多類別文本分類（情感檢測）問題。

✨ 主要特性

• 精準分類：在評估集上取得了出色的性能，能準確進行情感分類。
• 可配置性強：通過調整超參數，可以根據不同需求進行優化。

📚 詳細文檔

評估指標

該模型在評估集上取得了以下結果：

• 損失值（Loss）: 0.1613253802061081
• F1值（f1）: 0.9413321705151999

超參數

以下是模型訓練時使用的超參數：

{
    "epochs": 10,
    "train_batch_size": 16,
    "learning_rate": 3e-5, 
    "weight_decay":0.01,
    "load_best_model_at_end": true,
    "model_name":"roberta-base",
    "do_eval": True,
    "load_best_model_at_end":True
}

驗證指標

📄 許可證

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