cointegrated commited on
Commit
074a2d0
1 Parent(s): 4503599

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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
- ROC AUC 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
  |----------|------------|--------|--------|--------|--------|--------| --------| ----------------|
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
  |----------|------------|--------|--------|--------|--------|--------| --------| ----------------|