Update README.md
Browse files
README.md
CHANGED
|
@@ -74,16 +74,15 @@ Schizophrenia: 1131
|
|
| 74 |
Asperger: 1108
|
| 75 |
PTSD: 1089
|
| 76 |
|
| 77 |
-
The following hyperparameters were used during training:
|
| 78 |
|
| 79 |
model-finetuning: distilbert/distilbert-base-uncased
|
| 80 |
-
|
| 81 |
-
|
| 82 |
additional features (GoEmotions - SamLowe/roberta-base-go_emotions + SST2 - distilbert/distilbert-base-uncased-finetuned-sst-2-english):
|
| 83 |
negative, positive, admiration, amusement, anger, annoyance, approval, caring, confusion, curiosity,
|
| 84 |
desire, disappointment, disapproval, disgust, embarrassment, excitement, fear, gratitude, grief,
|
| 85 |
joy, love, nervousness, optimism, pride, realization, relief, remorse, sadness, surprise, neutral
|
| 86 |
|
|
|
|
| 87 |
|
| 88 |
learning_rate: 1e-5
|
| 89 |
train_batch_size: 64
|
|
|
|
| 74 |
Asperger: 1108
|
| 75 |
PTSD: 1089
|
| 76 |
|
|
|
|
| 77 |
|
| 78 |
model-finetuning: distilbert/distilbert-base-uncased
|
| 79 |
+
|
|
|
|
| 80 |
additional features (GoEmotions - SamLowe/roberta-base-go_emotions + SST2 - distilbert/distilbert-base-uncased-finetuned-sst-2-english):
|
| 81 |
negative, positive, admiration, amusement, anger, annoyance, approval, caring, confusion, curiosity,
|
| 82 |
desire, disappointment, disapproval, disgust, embarrassment, excitement, fear, gratitude, grief,
|
| 83 |
joy, love, nervousness, optimism, pride, realization, relief, remorse, sadness, surprise, neutral
|
| 84 |
|
| 85 |
+
The following hyperparameters were used during training:
|
| 86 |
|
| 87 |
learning_rate: 1e-5
|
| 88 |
train_batch_size: 64
|