kingabzpro commited on
Commit
ad4c3c8
1 Parent(s): e1e122e

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +30 -2
README.md CHANGED
@@ -14,5 +14,33 @@ metrics:
14
  - f1
15
  ---
16
 
17
- ## Gemma-2-9b-it-chat-doctor
18
- This model is a fine-tuned version of [meta-llama/Meta-Llama-3.1-8B-Instruct](https://huggingface.co/meta-llama/Meta-Llama-3.1-8B-Instruct) on an [suchintikasarkar/sentiment-analysis-for-mental-health](https://www.kaggle.com/datasets/suchintikasarkar/sentiment-analysis-for-mental-health) dataset.
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
14
  - f1
15
  ---
16
 
17
+ ## Llama-3.1-8B-Instruct-Mental-Health-Classification
18
+ This model is a fine-tuned version of [meta-llama/Meta-Llama-3.1-8B-Instruct](https://huggingface.co/meta-llama/Meta-Llama-3.1-8B-Instruct) on an [suchintikasarkar/sentiment-analysis-for-mental-health](https://www.kaggle.com/datasets/suchintikasarkar/sentiment-analysis-for-mental-health) dataset.
19
+
20
+ ## Results
21
+
22
+ 100%|██████████| 300/300 [03:24<00:00, 1.47it/s]
23
+ Accuracy: 0.913
24
+ Accuracy for label Normal: 0.972
25
+ Accuracy for label Depression: 0.913
26
+ Accuracy for label Anxiety: 0.667
27
+ Accuracy for label Bipolar: 0.800
28
+
29
+ **Classification Report:**
30
+ precision recall f1-score support
31
+
32
+ Normal 0.92 0.97 0.95 143
33
+ Depression 0.93 0.91 0.92 115
34
+ Anxiety 0.75 0.67 0.71 27
35
+ Bipolar 1.00 0.80 0.89 15
36
+
37
+ accuracy 0.91 300
38
+ macro avg 0.90 0.84 0.87 300
39
+ weighted avg 0.91 0.91 0.91 300
40
+
41
+
42
+ **Confusion Matrix:**
43
+ [[139 3 1 0]
44
+ [ 5 105 5 0]
45
+ [ 6 3 18 0]
46
+ [ 1 2 0 12]]