Update README.md

README.md

@@ -20,13 +20,12 @@ should probably proofread and complete it, then remove this comment. -->
 # distilhubert-finetuned-mixed-data
 
 This model is a fine-tuned version of [ntu-spml/distilhubert](https://huggingface.co/ntu-spml/distilhubert) on an unknown dataset.
-It achieves the following results on the evaluation set:
-- Loss: 0.8806
-- Accuracy: 0.7912
-- F1: 0.7772
-- Precision: 0.8022
-- Recall: 0.7912
-- Confusion Matrix: [[59, 1, 1, 2], [20, 35, 22, 0], [2, 7, 68, 0], [2, 0, 0, 54]]
+- Loss: 0.7808755040168762,
+- Accuracy: 0.8644688644688645,
+- F1: 0.8641694609590086,
+- Precision: 0.8653356589517041,
+- Recall: 0.8644688644688645,
+- Confusion Matrix: [[71, 9, 0, 3], [5, 42, 12, 0], [0, 7, 55, 0], [1, 0, 0, 68]]
 
 ## Model description
 
@@ -60,9 +59,9 @@ The following hyperparameters were used during training:
 
 ### Training results
 
-| Training Loss | Epoch   | Step | Validation Loss | Accuracy | F1     | Precision | Recall | Confusion Matrix                                               |
-|:-------------:|:-------:|:----:|:---------------:|:--------:|:------:|:---------:|:------:|:--------------------------------------------------------------:|
-| 0.4221        | 22.2222 | 100  | 0.8806          | 0.7912   | 0.7772 | 0.8022    | 0.7912 | [[59, 1, 1, 2], [20, 35, 22, 0], [2, 7, 68, 0], [2, 0, 0, 54]] |
+| Training Loss | Epoch   | Step | Validation Loss | Accuracy | F1     | Precision | Recall | Confusion Matrix                                              |
+|:-------------:|:-------:|:----:|:---------------:|:--------:|:------:|:---------:|:------:|:-------------------------------------------------------------:|
+| 0.5098        | 40.0000 | 50   | 0.7809          | 0.8645   | 0.8642 | 0.8653    | 0.8645 | [[71, 9, 0, 3], [5, 42, 12, 0], [0, 7, 55, 0], [1, 0, 0, 68]] |
 
 
 ### Framework versions