Kirkos27/news-category-dataset-model-distilbert

Model Card for Model ID

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