Update README.md
Browse files
README.md
CHANGED
@@ -77,6 +77,13 @@ PTSD: 1089
|
|
77 |
The following hyperparameters were used during training:
|
78 |
|
79 |
model-finetuning: distilbert/distilbert-base-uncased
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
80 |
|
81 |
learning_rate: 1e-5
|
82 |
train_batch_size: 64
|
|
|
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
|