update model card README.md

README.md

@@ -2,6 +2,8 @@
 license: apache-2.0
 tags:
 - generated_from_trainer
+datasets:
+- imdb
 model-index:
 - name: distilbert-base-uncased-finetuned-imdb
   results: []
@@ -12,9 +14,9 @@ should probably proofread and complete it, then remove this comment. -->
 
 # distilbert-base-uncased-finetuned-imdb
 
-This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on the None dataset.
+This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on the imdb dataset.
 It achieves the following results on the evaluation set:
-- Loss: 4.6041
+- Loss: 2.4721
 
 ## Model description
 
@@ -46,14 +48,14 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch | Step | Validation Loss |
 |:-------------:|:-----:|:----:|:---------------:|
-| 5.4216        | 1.0   | 10   | 4.7782          |
-| 4.8573        | 2.0   | 20   | 4.7080          |
-| 4.6729        | 3.0   | 30   | 4.5648          |
+| 2.7086        | 1.0   | 157  | 2.4897          |
+| 2.5796        | 2.0   | 314  | 2.4230          |
+| 2.5269        | 3.0   | 471  | 2.4354          |
 
 
 ### Framework versions
 
 - Transformers 4.18.0
 - Pytorch 1.11.0+cu113
-- Datasets 2.1.0
+- Datasets 2.2.0
 - Tokenizers 0.12.1