Update README.md

README.md

@@ -11,8 +11,8 @@ model-index:
 
 This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on the news_article_categorization dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.01338
-- F1_score(weighted): 1.0
+- Loss: 0.1548
+- F1_score(weighted): 0.96
 
 ### Model description
 
@@ -39,12 +39,10 @@ The following hyperparameters were used during training:
 - optimizer: AdamW(lr=1e-5 and epsilon=1e-08)
 - lr_scheduler_type: linear
 - lr_scheduler_warmup_steps: 0
-- num_epochs: 5
+- num_epochs: 2
 ### Training results
 | Training Loss | Epoch |  Validation Loss | f1 score   |
 |:-------------:|:-----:|:---------------: |:------:|
-| 0.5176        | 1.0   | 0.0466           | 0.9838 |
-| 0.0513        | 2.0   | 0.0051           | 1.0000 |
-| 0.0320        | 3.0   | 0.0032           | 1.0000 |
-| 0.0229        | 4.0   | 0.0018           | 1.0000 |
-| 0.0133        | 5.0   | 0.0017           | 1.0000 |
+| 0.6359        | 1.0   | 0.1739           | 0.9619 |
+| 0.1548        | 2.0   | 0.1898           | 0.9648 |
+