update model card README.md

README.md

@@ -22,10 +22,10 @@ model-index:
     metrics:
     - name: Accuracy
       type: accuracy
-      value: 0.5
+      value: 0.8150291876916989
     - name: F1
       type: f1
-      value: 0.5454545454545454
+      value: 0.794526570313788
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -35,9 +35,9 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [sentence-transformers/all-MiniLM-L6-v2](https://huggingface.co/sentence-transformers/all-MiniLM-L6-v2) on the quora dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.2761
-- Accuracy: 0.5
-- F1: 0.5455
+- Loss: 0.0865
+- Accuracy: 0.8150
+- F1: 0.7945
 
 ## Model description
 
@@ -57,22 +57,37 @@ More information needed
 
 The following hyperparameters were used during training:
 - learning_rate: 5e-05
-- train_batch_size: 8
-- eval_batch_size: 16
+- train_batch_size: 32
+- eval_batch_size: 64
 - seed: 42
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
-- num_epochs: 5
+- num_epochs: 20
 
 ### Training results
 
-| Training Loss | Epoch | Step | Validation Loss | Accuracy | F1     |
-|:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|
-| 0.2035        | 1.0   | 46   | 0.1972          | 0.3      | 0.4615 |
-| 0.1354        | 2.0   | 92   | 0.2265          | 0.45     | 0.5217 |
-| 0.0568        | 3.0   | 138  | 0.2754          | 0.45     | 0.5217 |
-| 0.0235        | 4.0   | 184  | 0.2770          | 0.45     | 0.5217 |
-| 0.0227        | 5.0   | 230  | 0.2761          | 0.5      | 0.5455 |
+| Training Loss | Epoch | Step   | Validation Loss | Accuracy | F1     |
+|:-------------:|:-----:|:------:|:---------------:|:--------:|:------:|
+| 0.087         | 1.0   | 11371  | 0.0829          | 0.4143   | 0.5535 |
+| 0.0794        | 2.0   | 22742  | 0.0783          | 0.6017   | 0.6458 |
+| 0.0606        | 3.0   | 34113  | 0.0756          | 0.3631   | 0.5327 |
+| 0.05          | 4.0   | 45484  | 0.0781          | 0.4475   | 0.5679 |
+| 0.0448        | 5.0   | 56855  | 0.0789          | 0.6856   | 0.6975 |
+| 0.0372        | 6.0   | 68226  | 0.0761          | 0.3922   | 0.5443 |
+| 0.033         | 7.0   | 79597  | 0.0786          | 0.7586   | 0.7494 |
+| 0.032         | 8.0   | 90968  | 0.0780          | 0.5011   | 0.5927 |
+| 0.0229        | 9.0   | 102339 | 0.0819          | 0.7513   | 0.7439 |
+| 0.0198        | 10.0  | 113710 | 0.0840          | 0.5522   | 0.6185 |
+| 0.0169        | 11.0  | 125081 | 0.0821          | 0.7959   | 0.7785 |
+| 0.0199        | 12.0  | 136452 | 0.0807          | 0.8353   | 0.8118 |
+| 0.0118        | 13.0  | 147823 | 0.0819          | 0.8418   | 0.8176 |
+| 0.0123        | 14.0  | 159194 | 0.0816          | 0.7577   | 0.7487 |
+| 0.0093        | 15.0  | 170565 | 0.0856          | 0.7934   | 0.7765 |
+| 0.0124        | 16.0  | 181936 | 0.0843          | 0.8484   | 0.8241 |
+| 0.008         | 17.0  | 193307 | 0.0838          | 0.7998   | 0.7818 |
+| 0.0106        | 18.0  | 204678 | 0.0872          | 0.8245   | 0.8027 |
+| 0.0066        | 19.0  | 216049 | 0.0857          | 0.8122   | 0.7922 |
+| 0.0059        | 20.0  | 227420 | 0.0865          | 0.8150   | 0.7945 |
 
 
 ### Framework versions