update model card README.md

README.md

@@ -3,7 +3,10 @@ license: apache-2.0
 tags:
 - generated_from_trainer
 metrics:
+- f1
 - accuracy
+- precision
+- recall
 model-index:
 - name: soft-search
   results: []
@@ -16,8 +19,11 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [distilbert-base-uncased-finetuned-sst-2-english](https://huggingface.co/distilbert-base-uncased-finetuned-sst-2-english) on the None dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.2610
-- Accuracy: 0.8333
+- Loss: 0.7833
+- F1: 0.5304
+- Accuracy: 0.6780
+- Precision: 0.5333
+- Recall: 0.5275
 
 ## Model description
 
@@ -46,18 +52,18 @@ The following hyperparameters were used during training:
 
 ### Training results
 
-| Training Loss | Epoch | Step | Validation Loss | Accuracy |
-|:-------------:|:-----:|:----:|:---------------:|:--------:|
-| No log        | 1.0   | 3    | 0.8268          | 0.75     |
-| No log        | 2.0   | 6    | 0.3217          | 0.8333   |
-| No log        | 3.0   | 9    | 0.2988          | 0.8333   |
-| 0.6789        | 4.0   | 12   | 0.2457          | 0.9167   |
-| 0.6789        | 5.0   | 15   | 0.2610          | 0.8333   |
+| Training Loss | Epoch | Step | Validation Loss | F1     | Accuracy | Precision | Recall |
+|:-------------:|:-----:|:----:|:---------------:|:------:|:--------:|:---------:|:------:|
+| 0.5776        | 1.0   | 50   | 0.6066          | 0.3803 | 0.6667   | 0.5294    | 0.2967 |
+| 0.5545        | 2.0   | 100  | 0.6261          | 0.4331 | 0.6629   | 0.5152    | 0.3736 |
+| 0.4599        | 3.0   | 150  | 0.7046          | 0.5472 | 0.6364   | 0.4793    | 0.6374 |
+| 0.2527        | 4.0   | 200  | 0.7285          | 0.5521 | 0.6742   | 0.5248    | 0.5824 |
+| 0.2423        | 5.0   | 250  | 0.7833          | 0.5304 | 0.6780   | 0.5333    | 0.5275 |
 
 
 ### Framework versions
 
-- Transformers 4.21.1
-- Pytorch 1.12.0+cu102
-- Datasets 2.4.0
-- Tokenizers 0.12.1
+- Transformers 4.24.0
+- Pytorch 1.13.0+cu117
+- Datasets 2.3.2
+- Tokenizers 0.13.2