Update README.md

README.md

@@ -7,29 +7,28 @@ tags:
 model-index:
 - name: IMDb_data_subset-MLM_with-custom_collator-distilbert-base-uncased
   results: []
+language:
+- en
 ---
 
-<!-- This model card has been generated automatically according to the information the Trainer had access to. You
-should probably proofread and complete it, then remove this comment. -->
-
 # IMDb_data_subset-MLM_with-custom_collator-distilbert-base-uncased
 
-This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on an unknown dataset.
+This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on IMDb dataset.
 It achieves the following results on the evaluation set:
 - Loss: 3.2538
 - Model Preparation Time: 0.0042
 
 ## Model description
 
-More information needed
+[distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased)
 
 ## Intended uses & limitations
 
-More information needed
+Mask filling
 
 ## Training and evaluation data
 
-More information needed
+The IMDb dataset is tokenized, and words are masked with 0.2 probability. The resulting dataset is downsampled, resulting in 10,000 training samples and 1,000 validation samples.
 
 ## Training procedure
 
@@ -61,4 +60,4 @@ The following hyperparameters were used during training:
 - Transformers 4.50.0
 - Pytorch 2.6.0+cu124
 - Datasets 3.4.1
-- Tokenizers 0.21.1
+- Tokenizers 0.21.1