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
- Downloads last month
- 4
This model does not have enough activity to be deployed to Inference API (serverless) yet. Increase its social
visibility and check back later, or deploy to Inference Endpoints (dedicated)
instead.