update model card README.md

README.md

@@ -19,11 +19,11 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.0005
-- Accuracy: 0.8043
-- F1: 0.8576
-- Precision: 0.7507
-- Recall: 1.0
+- Loss: 0.0006
+- Accuracy: 0.6309
+- F1: 0.7677
+- Precision: 0.6233
+- Recall: 0.9992
 
 ## Model description
 
@@ -49,18 +49,22 @@ The following hyperparameters were used during training:
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
 - lr_scheduler_warmup_steps: 500
-- num_epochs: 1
+- num_epochs: 5
 
 ### Training results
 
 | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1     | Precision | Recall |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|:---------:|:------:|
-| 0.0215        | 1.0   | 1956 | 0.0005          | 0.8043   | 0.8576 | 0.7507    | 1.0    |
+| 0.0176        | 1.0   | 1956 | 0.0009          | 0.9616   | 0.9695 | 0.9409    | 1.0    |
+| 0.0014        | 2.0   | 3912 | 0.0015          | 0.9864   | 0.9890 | 0.9783    | 1.0    |
+| 0.0011        | 3.0   | 5868 | 0.0008          | 0.7611   | 0.8363 | 0.7188    | 0.9996 |
+| 0.0008        | 4.0   | 7824 | 0.0008          | 0.7872   | 0.8514 | 0.7418    | 0.9992 |
+| 0.0006        | 5.0   | 9780 | 0.0006          | 0.6309   | 0.7677 | 0.6233    | 0.9992 |
 
 
 ### Framework versions
 
 - Transformers 4.18.0
 - Pytorch 1.10.0+cu111
-- Datasets 2.0.0
-- Tokenizers 0.11.6
+- Datasets 2.1.0
+- Tokenizers 0.12.1