Update README.md

README.md

@@ -12,24 +12,21 @@ probably proofread and complete it, then remove this comment. -->
 
 # distilbert_classifier_newsgroup
 
-This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on an unknown dataset.
-It achieves the following results on the evaluation set:
+This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on the [20 Newsgroups](http://qwone.com/~jason/20Newsgroups/) data set is a collection of approximately 20,000 newsgroup documents, partitioned (nearly) evenly across 20 different newsgroups.
+It achieves the following results on the evaluation set: loss: 0.5660 - accuracy: 0.8371
 
 
-## Model description
-
-More information needed
-
-## Intended uses & limitations
-
-More information needed
 
 ## Training and evaluation data
 
-More information needed
+- The training set contained 10182 rows with features - text and label.
+- The evaluation set contained 7532 rows with features - text and label.
 
 ## Training procedure
-
+- Setup the model checkpoint as [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased)
+- Intialize the starting weights with the model checkpoint and give it the number of labels - i.e., 20 in this case.
+- Will be training for 3 epochs
+- Using a batch size of 16
 ### Training hyperparameters
 
 The following hyperparameters were used during training:
@@ -38,7 +35,7 @@ The following hyperparameters were used during training:
 
 ### Training results
 
-
+- Results on the evaluation set: loss: 0.5660 - accuracy: 0.8371
 
 ### Framework versions