Update README.md

README.md

@@ -26,16 +26,16 @@ model-index:
       split: validation
     metrics:
     - type: accuracy
-      value: 0.6710097719869706
+      value: 0.6742671009771987
       name: Accuracy
     - type: f1
-      value: 0.6576271186440678
+      value: 0.6710526315789473
       name: F1
     - type: precision
-      value: 0.6830985915492958
+      value: 0.6754966887417219
       name: Precision
     - type: recall
-      value: 0.6339869281045751
+      value: 0.6666666666666666
       name: Recall
 ---
 
@@ -69,7 +69,7 @@ with torch.inference_mode():
     )
     p = torch.softmax(prediction.logits, -1).cpu().numpy()[0]
 print({v: p[k] for k, v in model.config.id2label.items()})
-# {'not_entailment': 0.61647415, 'entailment': 0.38352585}
+# {'not_entailment': 0.68763, 'entailment': 0.31237}
 
 # An example concerning sentiments
 premise2 = 'Я ненавижу желтые занавески'
@@ -80,7 +80,7 @@ with torch.inference_mode():
     )
     p = torch.softmax(prediction.logits, -1).cpu().numpy()[0]
 print({v: p[k] for k, v in model.config.id2label.items()})
-# {'not_entailment': 0.520086, 'entailment': 0.47991407}
+# {'not_entailment': 0.558041, 'entailment': 0.441959}
 ```
 
 ## Model Performance Metrics
@@ -89,13 +89,13 @@ The following metrics summarize the performance of the model on the test dataset
 
 | Metric                           | Value                     |
 |----------------------------------|---------------------------|
-| **Validation Loss**              | 0.6546                    |
-| **Validation Accuracy**          | 67.10%                    |
-| **Validation F1 Score**          | 65.76%                    |
-| **Validation Precision**         | 68.31%                    |
-| **Validation Recall**            | 63.40%                    |
-| **Validation Runtime***          | 0.2546 seconds            |
-| **Samples per Second***          | 1 205.80                    |
-| **Steps per Second***            | 7.86                     |
+| **Validation Loss**              | 0.6492                    |
+| **Validation Accuracy**          | 67.43%                    |
+| **Validation F1 Score**          | 67.11%                    |
+| **Validation Precision**         | 67.55%                    |
+| **Validation Recall**            | 66.67%                    |
+| **Validation Runtime***          | 0.2631 seconds            |
+| **Samples per Second***          | 1 167.02                    |
+| **Steps per Second***            | 7.60                     |
 
 *Using T4 GPU with Google Colab