Text Classification
Transformers
Safetensors
roberta
mental-health
multi-label
mental-roberta-base
text-embeddings-inference
Instructions to use mahir2021/mental-roberta-depEMO with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use mahir2021/mental-roberta-depEMO with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="mahir2021/mental-roberta-depEMO")# pip install -U transformers accelerate # Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("mahir2021/mental-roberta-depEMO") model = AutoModelForSequenceClassification.from_pretrained("mahir2021/mental-roberta-depEMO", device_map="auto") - Notebooks
- Google Colab
- Kaggle
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