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
|