mdsajjadullah commited on
Commit
3e889ce
Β·
verified Β·
1 Parent(s): 6a493ef

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +53 -3
README.md CHANGED
@@ -1,3 +1,53 @@
1
- ---
2
- license: mit
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ Fine-tuned DistilBERT for binary depression classification from
2
+ social media text (Twitter + Reddit).
3
+
4
+ Trained as part of the ExplainDepression Pipeline β€” a research
5
+ system for explainable and fair depression detection using SHAP
6
+ attributions, Depression Explanation Score (DES), and Explanation
7
+ Consistency Index across Demographics (ECID).
8
+
9
+ ━━━━━━━━━━━━━━━━━━━━━━━━━━━
10
+
11
+ PERFORMANCE
12
+ Accuracy β†’ 96.17%
13
+ F1 Score β†’ 0.9617
14
+ AUC-ROC β†’ 0.9937
15
+ ECID β†’ 0.2063 (bias detected in 3 clinical categories)
16
+
17
+ ━━━━━━━━━━━━━━━━━━━━━━━━━━━
18
+
19
+ TRAINING DETAILS
20
+ Base Model β†’ distilbert-base-uncased
21
+ Epochs β†’ 3
22
+ Batch Size β†’ 64
23
+ Learning Rate β†’ 3e-5
24
+ Max Length β†’ 96 tokens
25
+ Train Size β†’ 11,200 posts
26
+ Platform β†’ Kaggle GPU T4
27
+
28
+ ━━━━━━━━━━━━━━━━━━━━━━━━━━━
29
+
30
+ LABELS
31
+ LABEL_0 β†’ Not Depressed
32
+ LABEL_1 β†’ Depressed
33
+
34
+ ━━━━━━━━━━━━━━━━━━━━━━━━━━━
35
+
36
+ HOW TO USE
37
+
38
+ from transformers import pipeline
39
+
40
+ clf = pipeline(
41
+ "text-classification",
42
+ model="mdsajjadullah/explainDepression-distilbert",
43
+ return_all_scores=True
44
+ )
45
+
46
+ result = clf("I feel hopeless and completely alone.")
47
+ # LABEL_1 score = depression probability
48
+
49
+ ━━━━━━━━━━━━━━━━━━━━━━━━━━━
50
+
51
+ INTENDED FOR RESEARCH ONLY.
52
+ Not for clinical diagnosis or deployment without
53
+ professional mental health oversight.