Update README.md
Browse files
README.md
CHANGED
|
@@ -1,3 +1,18 @@
|
|
| 1 |
---
|
| 2 |
license: apache-2.0
|
| 3 |
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
---
|
| 2 |
license: apache-2.0
|
| 3 |
---
|
| 4 |
+
|
| 5 |
+
# PsychBERT Fine-Tuned on Simple Multi-class Problem
|
| 6 |
+
|
| 7 |
+
This is a version of https://huggingface.co/mnaylor/psychbert-cased which was fine-tuned to illustrate performance on a multi-class classification problem
|
| 8 |
+
involving the detection of different types of language relating to mental health. The classes are as follows:
|
| 9 |
+
|
| 10 |
+
* 0: Negative / unrelated to mental health
|
| 11 |
+
* 1: Mental illnesses
|
| 12 |
+
* 2: Anxiety
|
| 13 |
+
* 3: Depression
|
| 14 |
+
* 4: Social anxiety
|
| 15 |
+
* 5: Loneliness
|
| 16 |
+
|
| 17 |
+
The dataset for this model was taken from Reddit and Twitter, and labels were assigned based on the post appearing in certain subreddits or containing certain hashtags.
|
| 18 |
+
For more information, see the [PsychBERT paper](https://ieeexplore.ieee.org/document/9669469).
|