Mental-RoBERTa

Multi-label emotion classifier trained on depression-related text.

  • Base model: mental/mental-roberta-base
  • Architecture: Standard AutoModelForSequenceClassification
  • Labels (8): anger, brain dysfunction (forget), emptiness, hopelessness, loneliness, sadness, suicide intent, worthlessness
  • Seed: 42, Epochs: 5, LR: 2e-05, MaxLen: 256

Test metrics

{
  "eval_loss": 0.711999237537384,
  "eval_P-Mic": 0.8185206422018348,
  "eval_R-Mic": 0.8535127055306427,
  "eval_F1-Mic": 0.8356505195375384,
  "eval_P-Mac": 0.7865680451631196,
  "eval_R-Mac": 0.7994304832879919,
  "eval_F1-Mac": 0.785985179313559
}
Downloads last month
66
Safetensors
Model size
0.1B params
Tensor type
F32
ยท
Inference Providers NEW
This model isn't deployed by any Inference Provider. ๐Ÿ™‹ Ask for provider support