Update README.md
Browse files
README.md
CHANGED
|
@@ -22,15 +22,16 @@ It achieves the following results on the evaluation set:
|
|
| 22 |
|
| 23 |
## Model description
|
| 24 |
|
| 25 |
-
|
| 26 |
|
| 27 |
-
|
| 28 |
|
| 29 |
-
More information needed
|
| 30 |
|
| 31 |
-
##
|
|
|
|
| 32 |
|
| 33 |
-
|
|
|
|
| 34 |
|
| 35 |
## Training procedure
|
| 36 |
|
|
|
|
| 22 |
|
| 23 |
## Model description
|
| 24 |
|
| 25 |
+
This model is a fine-tuned version of the DistilBERT model for sequence classification tasks. It was trained using Hugging Face's transformers and TensorFlow. The model expects input sequences to be tokenized according to the DistilBERT's tokenizer.
|
| 26 |
|
| 27 |
+
The model was trained specifically for classifying text into 20 different categories derived from the 20 Newsgroups dataset. These categories include various topics such as 'alt.atheism', 'comp.graphics', 'comp.os.ms-windows.misc', 'comp.sys.ibm.pc.hardware', 'comp.sys.mac.hardware', 'comp.windows.x', 'misc.forsale', 'rec.autos', 'rec.motorcycles', 'rec.sport.baseball', 'rec.sport.hockey', 'sci.crypt', 'sci.electronics', 'sci.med', 'sci.space', 'soc.religion.christian', 'talk.politics.guns', 'talk.politics.mideast', 'talk.politics.misc', 'talk.religion.misc'.
|
| 28 |
|
|
|
|
| 29 |
|
| 30 |
+
## Intended uses & limitations
|
| 31 |
+
This model is intended for classifying text into the above mentioned 20 categories. It can be used for categorizing text data from similar domains or topics.
|
| 32 |
|
| 33 |
+
## Training and evaluation data
|
| 34 |
+
the model was trained on 90% of the data from the 20 Newsgroups dataset, with the remaining 10% used for validation.
|
| 35 |
|
| 36 |
## Training procedure
|
| 37 |
|