Update README.md

README.md

@@ -16,15 +16,15 @@ probably proofread and complete it, then remove this comment. -->
 
 # apriadiazriel/bert-cased-jnlpba
 
-This model is a fine-tuned version of [bert-base-cased](https://huggingface.co/bert-base-cased) on the [JNLPBA](https://huggingface.co/datasets/jnlpba/jnlpba).
+This model is a fine-tuned version of [bert-base-cased](https://huggingface.co/bert-base-cased) on the [JNLPBA](https://huggingface.co/datasets/jnlpba/jnlpba) dataset.
 It achieves the following results on the evaluation set:
 - Train Loss: 0.0851
 - Validation Loss: 0.2221
-- Train Precision: 0.6744
-- Train Recall: 0.7808
-- Train F1: 0.7237
-- Train Accuracy: 0.9371
-- Epoch: 4
+- Precision: 0.6744
+- Recall: 0.7808
+- F1: 0.7237
+- Accuracy: 0.9371
+- Epoch: 5
 
 ## Model description
 
@@ -48,7 +48,7 @@ The following hyperparameters were used during training:
 
 ### Training results
 
-| Train Loss | Validation Loss | Train Precision | Train Recall | Train F1 | Train Accuracy | Epoch |
+| Train Loss | Validation Loss | Precision | Recall | F1 | Accuracy | Epoch |
 |:----------:|:---------------:|:---------------:|:------------:|:--------:|:--------------:|:-----:|
 | 0.2424     | 0.1998          | 0.6507          | 0.7606       | 0.7014   | 0.9322         | 0     |
 | 0.1426     | 0.1975          | 0.6613          | 0.7832       | 0.7171   | 0.9364         | 1     |