Commit
·
958dfb3
1
Parent(s):
47da2f1
Update README.md
Browse files
README.md
CHANGED
|
@@ -25,6 +25,8 @@ The model has been trained with Adam optimizer for 40 epochs with learning rate
|
|
| 25 |
|
| 26 |
ROC AUC of the predicted probabilities on the test dataset is the following:
|
| 27 |
|
| 28 |
-
| label
|
| 29 |
-
|
| 30 |
-
| AUC
|
|
|
|
|
|
|
|
|
| 25 |
|
| 26 |
ROC AUC of the predicted probabilities on the test dataset is the following:
|
| 27 |
|
| 28 |
+
| label | no emotion | joy |sadness |surprise| fear |anger | mean |
|
| 29 |
+
|----------|------------|--------|--------|--------|--------|--------| --------|
|
| 30 |
+
| AUC | 0.9406 | 0.9518 | 0.9372 | 0.8634 | 0.9663 | 0.6761 | 0.8892 |
|
| 31 |
+
| F1 micro | 0.8624 | 0.9389 | 0.9362 | 0.9469 | 0.9575 | 0.9261 | 0.9280 |
|
| 32 |
+
| F1 macro | 0.8562 | 0.8962 | 0.9017 | 0.8366 | 0.8359 | 0.6820 | 0.8348 |
|