Downloads · 30 days
28
10% of all-time downloads
sajjadhadi/Mental-Disorder-Detection-DeepSeek-R1-Distill-Qwen-1.5B-v1
Mental-Disorder-Detection-DeepSeek-R1-Distill-Qwen-1.5B-v1 is a machine learning model from sajjadhadi. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for peft. The card lists the license as mit.
should probably proofread and complete it, then remove this comment. --
Downloads · 30 days
28
10% of all-time downloads
All-time downloads
293
Public
Repo size
772 MB
Likes
0
Public
Click a slice to open those files.
.arrow480 MB · 96%
From the Hugging Face model README
This mental disorder detection model is designed for research and educational purposes only. It is not intended for use in real-world medical environments or to detect actual mental health conditions. For any medical concerns, always consult a qualified healthcare professional. The developers assume no responsibility for misuse in clinical or real-life scenarios.
This model is a fine-tuned version of deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B.
This Large Language Model (LLM), based on DeepSeek R1 Distill Qwen with 1.5 billion parameters, is fine-tuned on a dataset of 48,000 records to classify five distinct mental health-related classes: depression, suicide, bipolar, anxiety, and expressing. The model leverages lightweight quantization techniques (4-bit) for efficient computation while maintaining accuracy. Though designed for experimental and research purposes, it demonstrates the potential to assist in identifying mental health patterns through text, offering valuable insights for educational and academic applications.
| Class | Top-1 Accuracy | Top-1 Count | Top-2 Accuracy | Top-2 Count |
|---|---|---|---|---|
| Overall | 0.62 | 1526/2450 | 0.79 | 1946/2450 |
| Depression | 0.56 | 478/861 | 0.87 | 752/861 |
| Suicide | 0.63 | 294/466 | 0.76 | 352/466 |
| Bipolar | 0.71 | 248/351 | 0.75 | 264/351 |
| Anxiety | 0.67 | 277/413 | 0.73 | 303/413 |
| Expressing | 0.64 | 229/359 | 0.77 | 275/359 |
For more information about the usage and metrics you can check this Kaggle notebook.