Update README.md
Browse files
README.md
CHANGED
|
@@ -21,7 +21,7 @@ model-index:
|
|
| 21 |
widget:
|
| 22 |
- text: "I have trouble understanding what other people think or feel. I also like numbers, and finding patterns in numbers."
|
| 23 |
---
|
| 24 |
-
This model is a fine-tuned version of distilbert-base-uncased on Reddit dataset contains text related to mental health reports of users. it predicts mental health disorders from textual content.
|
| 25 |
|
| 26 |
It achieves the following results on the validation set:
|
| 27 |
|
|
@@ -31,8 +31,10 @@ It achieves the following results on the validation set:
|
|
| 31 |
* Precision: 0.7671
|
| 32 |
|
| 33 |
# Description
|
| 34 |
-
This model is based on an existing lighter variation of BERT (distilBERT), in order to predict different mental disorders.
|
| 35 |
-
|
|
|
|
|
|
|
| 36 |
|
| 37 |
It includes the following classes:
|
| 38 |
|
|
|
|
| 21 |
widget:
|
| 22 |
- text: "I have trouble understanding what other people think or feel. I also like numbers, and finding patterns in numbers."
|
| 23 |
---
|
| 24 |
+
This model is a hybrid fine-tuned version of distilbert-base-uncased on Reddit dataset contains text related to mental health reports of users. it predicts mental health disorders from textual content.
|
| 25 |
|
| 26 |
It achieves the following results on the validation set:
|
| 27 |
|
|
|
|
| 31 |
* Precision: 0.7671
|
| 32 |
|
| 33 |
# Description
|
| 34 |
+
This model is based on an existing lighter variation of BERT (distilBERT), in order to predict different mental disorders.
|
| 35 |
+
* It is using combinded features of sentiments and emotions (distilbert-base-uncased-finetuned-sst-2-english and roberta-base-go_emotions).
|
| 36 |
+
* It is trained on a costume dataset of texts or posts (from Reddit) about general experiences of users with mental health problems.
|
| 37 |
+
* All direct mentions of the disorder names in the texts were removed.
|
| 38 |
|
| 39 |
It includes the following classes:
|
| 40 |
|