kingabzpro
commited on
Commit
•
ad4c3c8
1
Parent(s):
e1e122e
Update README.md
Browse files
README.md
CHANGED
@@ -14,5 +14,33 @@ metrics:
|
|
14 |
- f1
|
15 |
---
|
16 |
|
17 |
-
##
|
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]]
|