update model card README.md

README.md

@@ -1,5 +1,5 @@
 ---
-license: apache-2.0
+license: mit
 tags:
 - generated_from_trainer
 metrics:
@@ -17,13 +17,13 @@ should probably proofread and complete it, then remove this comment. -->
 
 # sentence-compression
 
-This model is a fine-tuned version of [distilbert-base-cased](https://huggingface.co/distilbert-base-cased) on an unknown dataset.
+This model is a fine-tuned version of [roberta-base](https://huggingface.co/roberta-base) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.4221
-- Accuracy: 0.8121
-- F1: 0.7275
-- Precision: 0.7317
-- Recall: 0.7233
+- Loss: 0.3054
+- Accuracy: 0.8802
+- F1: 0.7439
+- Precision: 0.7661
+- Recall: 0.7229
 
 ## Model description
 
@@ -55,13 +55,14 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1     | Precision | Recall |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|:---------:|:------:|
-| 0.6263        | 1.0   | 50   | 0.6252          | 0.6549   | 0.0183 | 0.6667    | 0.0093 |
-| 0.4727        | 2.0   | 100  | 0.4900          | 0.7805   | 0.6472 | 0.7309    | 0.5807 |
-| 0.4053        | 3.0   | 150  | 0.4221          | 0.8121   | 0.7275 | 0.7317    | 0.7233 |
+| 0.317         | 1.0   | 200  | 0.3566          | 0.8513   | 0.6227 | 0.7998    | 0.5099 |
+| 0.346         | 2.0   | 400  | 0.5853          | 0.7919   | 0.6724 | 0.5413    | 0.8875 |
+| 0.2365        | 3.0   | 600  | 0.3054          | 0.8802   | 0.7439 | 0.7661    | 0.7229 |
 
 
 ### Framework versions
 
 - Transformers 4.12.5
 - Pytorch 1.10.0+cu111
+- Datasets 1.16.1
 - Tokenizers 0.10.3