This model is a finetuning of the distilbert/distilbert-base-uncased llm for the news-category-dataset.
Model Details
Training took ~1 hour on google colab using an L4 GPU. For the dataset presenting 42 labels, it provides the following accuracies:
{'eval_loss': 1.003881573677063,
'eval_accuracy': 0.7176901049952275,
}
Model Description
This is the model card of a 🤗 transformers model that has been pushed on the Hub. This model card has been automatically generated.
Developed by: [Kirkos27]
Model type: Text Classification
Language(s) (NLP): English
Finetuned from model : distilbert/distilbert-base-uncased
