update model card README.md

README.md

@@ -1,7 +1,9 @@
 ---
-license: apache-2.0
+license: mit
 tags:
 - generated_from_trainer
+metrics:
+- f1
 model-index:
 - name: Greg-Sentiment-classifier
   results: []
@@ -12,7 +14,10 @@ should probably proofread and complete it, then remove this comment. -->
 
 # Greg-Sentiment-classifier
 
-This model is a fine-tuned version of [distilbert-base-cased-distilled-squad](https://huggingface.co/distilbert-base-cased-distilled-squad) on an unknown dataset.
+This model is a fine-tuned version of [microsoft/MiniLM-L12-H384-uncased](https://huggingface.co/microsoft/MiniLM-L12-H384-uncased) on an unknown dataset.
+It achieves the following results on the evaluation set:
+- Loss: 1.0165
+- F1: 0.3222
 
 ## Model description
 
@@ -31,7 +36,7 @@ More information needed
 ### Training hyperparameters
 
 The following hyperparameters were used during training:
-- learning_rate: 2e-05
+- learning_rate: 0.004
 - train_batch_size: 16
 - eval_batch_size: 16
 - seed: 42
@@ -39,6 +44,18 @@ The following hyperparameters were used during training:
 - lr_scheduler_type: linear
 - num_epochs: 6
 
+### Training results
+
+| Training Loss | Epoch | Step | Validation Loss | F1     |
+|:-------------:|:-----:|:----:|:---------------:|:------:|
+| 1.0386        | 1.0   | 499  | 1.0199          | 0.2347 |
+| 1.0437        | 2.0   | 998  | 1.0226          | 0.2347 |
+| 1.066         | 3.0   | 1497 | 1.0207          | 0.2347 |
+| 1.0183        | 4.0   | 1996 | 1.0173          | 0.3222 |
+| 0.9965        | 5.0   | 2495 | 1.0194          | 0.2347 |
+| 0.9548        | 6.0   | 2994 | 1.0165          | 0.3222 |
+
+
 ### Framework versions
 
 - Transformers 4.30.2