update model card README.md

README.md

@@ -4,6 +4,8 @@ tags:
 - generated_from_trainer
 metrics:
 - accuracy
+- precision
+- recall
 model-index:
 - name: product_classifier
   results: []
@@ -16,8 +18,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.6778
-- Accuracy: 0.8009
+- Loss: 0.6760
+- Accuracy: {'accuracy': 0.80125}
+- Precision: {'precision': 0.785989926719994}
+- Recall: {'recall': 0.7755906520102293}
+- F1 Score: {'f1': 0.7704315421053631}
 
 ## Model description
 
@@ -46,9 +51,9 @@ The following hyperparameters were used during training:
 
 ### Training results
 
-| Training Loss | Epoch | Step | Validation Loss | Accuracy |
-|:-------------:|:-----:|:----:|:---------------:|:--------:|
-| 0.9356        | 1.0   | 3200 | 0.6731          | 0.8027   |
+| Training Loss | Epoch | Step | Validation Loss | Accuracy                | Precision                         | Recall                         | F1 Score                  |
+|:-------------:|:-----:|:----:|:---------------:|:-----------------------:|:---------------------------------:|:------------------------------:|:-------------------------:|
+| 0.9575        | 1.0   | 3200 | 0.6832          | {'accuracy': 0.7978125} | {'precision': 0.7851098622896849} | {'recall': 0.7737991362724596} | {'f1': 0.771520016712035} |
 
 
 ### Framework versions