Downloads · 30 days
0
mentalbert-multiclass-das
alfiyahqthz/mentalbert-multiclass-das
mentalbert-multiclass-das is a machine learning model from alfiyahqthz. Use it for the machine learning task on the model card, and read the license before you ship it in a product. The card lists the license as mit.
Fine-tuned for Depression/Anxiety/Stress classification.
Access
Public
Updated Feb 10, 2026
Repo size
878 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.bin439 MB · 100%
At a glance
- License
- mit
- Access
- Public
- Created
- Feb 10, 2026
- Updated
- Feb 10, 2026
- SHA
- df122ca3
Model card
From the Hugging Face model README
Mental Health Multiclass Classification
Fine-tuned for Depression/Anxiety/Stress classification.
Performance
- Validation Loss: 0.2910
- Validation Accuracy: 0.8893
- F1 (Macro): 0.8819
Usage
import torch
from transformers import AutoTokenizer
model = torch.load("pytorch_model.bin")
tokenizer = AutoTokenizer.from_pretrained("alfiyahqthz/mentalbert-multiclass-das")
text = "I feel depressed"
inputs = tokenizer(text, return_tensors="pt", padding=True, truncation=True)
logits = model(inputs['input_ids'], inputs['attention_mask'])
prediction = logits.argmax(dim=1).item()
# 0=Depression, 1=Anxiety, 2=Stress
- License
- mit
- Languages
- en
- Created
- Feb 10, 2026
- Updated
- Feb 10, 2026