cointegrated
commited on
Commit
·
074a2d0
1
Parent(s):
4503599
Update README.md
Browse files
README.md
CHANGED
@@ -23,7 +23,7 @@ The model on the [CEDR dataset](https://huggingface.co/datasets/cedr) described
|
|
23 |
|
24 |
The model has been trained with Adam optimizer for 40 epochs with learning rate `1e-5` and batch size 64 [in this notebook](https://colab.research.google.com/drive/1AFW70EJaBn7KZKRClDIdDUpbD46cEsat?usp=sharing).
|
25 |
|
26 |
-
|
27 |
|
28 |
| label | no emotion | joy |sadness |surprise| fear |anger | mean | mean (emotions) |
|
29 |
|----------|------------|--------|--------|--------|--------|--------| --------| ----------------|
|
|
|
23 |
|
24 |
The model has been trained with Adam optimizer for 40 epochs with learning rate `1e-5` and batch size 64 [in this notebook](https://colab.research.google.com/drive/1AFW70EJaBn7KZKRClDIdDUpbD46cEsat?usp=sharing).
|
25 |
|
26 |
+
The quality of the predicted probabilities on the test dataset is the following:
|
27 |
|
28 |
| label | no emotion | joy |sadness |surprise| fear |anger | mean | mean (emotions) |
|
29 |
|----------|------------|--------|--------|--------|--------|--------| --------| ----------------|
|