Downloads · 30 days
6
2% of all-time downloads
news-category-dataset-model-distilbert
Kirkos27/news-category-dataset-model-distilbert
news-category-dataset-model-distilbert is a text classification model from Kirkos27. Use it when you need a label for a piece of text. It is set up for transformers. The card lists the license as mit.
This model is a finetuning of the distilbert/distilbert-base-uncased llm for the news-category-dataset.
All-time downloads
387
Public
Parameters
67M
268 MB on disk
Likes
0
Public
Repo makeup
Click a slice to open those files.
.safetensors268 MB · 100%
At a glance
- Task
- Text Classification
- Library
- transformers
- License
- mit
- Model type
- distilbert
- Access
- Public
- Created
- Jul 24, 2024
- Updated
- Jul 24, 2024
- SHA
- 282dff82
Try a prompt
Datasets
Model card
From the Hugging Face model README
Model Card for Model ID
<!-- Provide a quick summary of what the model is/does. -->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
<!-- Provide a longer summary of what this model is. -->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
- Task
- Text Classification
- Library
- transformers
- Type
- distilbert
- License
- mit
- Languages
- en
- Created
- Jul 24, 2024
- Updated
- Jul 24, 2024