Update README.md

README.md

@@ -30,10 +30,10 @@ This named entity recognition model detects negation and speculation entities, a
 - Speculated: speculated entity or event (e.g. *posiblemente **sobreviva***)
 
 The model achieves the following results on the test set (when trained with the training and development set; results are averaged over 5 evaluation rounds):
-- Precision: 0.833 (±0.001)
-- Recall: 0.870 (±0.001)
-- F1: 0.851 (±0.001)
-- Accuracy: 0.956 (±0.001)
+- Precision: 0.838 (±0.003)
+- Recall: 0.866 (±0.005)
+- F1: 0.852 (±0.003)
+- Accuracy: 0.986 (±0.001)
 
 ## Model description
 
@@ -100,24 +100,24 @@ The following hyperparameters were used during training:
 - seed: we used different seeds for 5 evaluation rounds, and uploaded the model with the best results
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
-- num_epochs: 4
+- num_epochs: 8
 
 
 ### Training results (test set; average and standard deviation of 5 rounds with different seeds)
 
 |   Precision    |     Recall     |       F1       |    Accuracy    |
 |:--------------:|:--------------:|:--------------:|:--------------:|
-| 0.833 (±0.001) | 0.870 (±0.001) | 0.851 (±0.001) | 0.986 (±0.001) |
+| 0.838 (±0.003) | 0.866 (±0.005) | 0.852 (±0.003) | 0.986 (±0.001) |
 
 
 **Results per class (test set; average and standard deviation of 5 rounds with different seeds)**  
 
 | Class       |   Precision    |     Recall     |       F1       |  Support  |
 |:-----------:|:--------------:|:--------------:|:--------------:|:---------:|
-| Neg_cue     | 0.944 (±0.001) | 0.963 (±0.002) | 0.954 (±0.001) |    2416   |
-| Negated     | 0.805 (±0.003) | 0.843 (±0.005) | 0.823 (±0.003) |    3064   |
-| Spec_cue    | 0.800 (±0.005) | 0.862 (±0.006) | 0.830 (±0.002) |     746   |
-| Speculated  | 0.683 (±0.009) | 0.735 (±0.010) | 0.708 (±0.009) |     993   |
+| Neg_cue     | 0.945 (±0.004) | 0.961 (±0.002) | 0.953 (±0.003) |    2416   |
+| Negated     | 0.815 (±0.003) | 0.838 (±0.005) | 0.826 (±0.003) |    3064   |
+| Spec_cue    | 0.811 (±0.005) | 0.868 (±0.009) | 0.839 (±0.005) |     746   |
+| Speculated  | 0.685 (±0.009) | 0.719 (±0.016) | 0.701 (±0.008) |     993   |
 
 
 ### Framework versions