update model card README.md

README.md

@@ -2,24 +2,11 @@
 license: mit
 tags:
 - generated_from_trainer
-datasets:
-- sst2
 metrics:
 - accuracy
 model-index:
 - name: sentiment-model-saagie
-  results:
-  - task:
-      name: Text Classification
-      type: text-classification
-    dataset:
-      name: sst2
-      type: sst2
-      args: default
-    metrics:
-    - name: Accuracy
-      type: accuracy
-      value: 0.79
+  results: []
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -27,10 +14,10 @@ should probably proofread and complete it, then remove this comment. -->
 
 # sentiment-model-saagie
 
-This model is a fine-tuned version of [prajjwal1/bert-tiny](https://huggingface.co/prajjwal1/bert-tiny) on the sst2 dataset.
+This model is a fine-tuned version of [prajjwal1/bert-tiny](https://huggingface.co/prajjwal1/bert-tiny) on the None dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.5486
-- Accuracy: 0.79
+- Loss: 0.5490
+- Accuracy: 0.7817
 
 ## Model description
 
@@ -61,9 +48,9 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch | Step | Validation Loss | Accuracy |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|
-| 0.5257        | 1.0   | 1500 | 0.4729          | 0.7733   |
-| 0.3846        | 2.0   | 3000 | 0.5373          | 0.7783   |
-| 0.3441        | 3.0   | 4500 | 0.5486          | 0.79     |
+| 0.528         | 1.0   | 1500 | 0.4654          | 0.77     |
+| 0.3824        | 2.0   | 3000 | 0.5333          | 0.7733   |
+| 0.337         | 3.0   | 4500 | 0.5490          | 0.7817   |
 
 
 ### Framework versions