CoolFace
Modelpublic

adicadi/medipal-mental-health

sourceHugging Facemitupdated 10mo agoView on Hugging Face
0likes15downloads
Model Card

MediPal Mental Health Classification Model

Model Description

This is a fine-tuned RoBERTa-Large model for mental health sentiment classification. It classifies text into 7 categories:

  • —Normal: Healthy mental state
  • —Anxiety: Anxious thoughts and worry
  • —Depression: Depressive symptoms
  • —Stress: Stress-related content
  • —Bipolar: Bipolar disorder indicators
  • —Personality disorder: Personality disorder symptoms
  • —Suicidal: Suicidal ideation (requires immediate intervention)

Model Performance

  • —Accuracy: 76.5%
  • —F1 Score: 76.6%
  • —Base Model: RoBERTa-Large
  • —Training Dataset: Mental health text corpus

Intended Use

This model is designed for mental health journal analysis and should be used as a supportive tool, not as a replacement for professional mental health diagnosis.

How to Use

Using Transformers

from transformers import RobertaTokenizer, RobertaForSequenceClassification import torch

model = RobertaForSequenceClassification.frompretrained("adicadi/medipal-mental-health") tokenizer = RobertaTokenizer.frompretrained("adicadi/medipal-mental-health")

text = "I feel anxious and worried" inputs = tokenizer(text, returntensors="pt", padding=True, truncation=True, maxlength=128)

with torch.nograd(): outputs = model(**inputs) predictions = torch.nn.functional.softmax(outputs.logits, dim=-1) predictedclass = torch.argmax(predictions).item()

labels = ['Anxiety', 'Bipolar', 'Depression', 'Normal', 'Personality disorder', 'Stress', 'Suicidal'] print(f"Predicted: {labels[predictedclass]}") print(f"Confidence: {predictions[predictedclass].item():.2f}")

Using Inference API

curl https://api-inference.huggingface.co/models/adicadi/medipal-mental-health -H "Authorization: Bearer YOURHFTOKEN" -H "Content-Type: application/json" -d '{"inputs": "I feel anxious and worried"}'

Limitations

  • —Not a substitute for professional mental health diagnosis
  • —May not capture cultural or linguistic nuances
  • —Trained on English text only
  • —Requires professional validation for clinical use

Ethical Considerations

This model deals with sensitive mental health data. Users should:

  • —Handle predictions with care and empathy
  • —Provide appropriate crisis resources for high-risk predictions
  • —Not use as the sole basis for mental health decisions
  • —Comply with local healthcare regulations

Contact

For questions or issues, please contact the model author.