Downloads · 30 days
9
2% of all-time downloads
agnews_distilbert_finetuned
billster45/agnews_distilbert_finetuned
agnews_distilbert_finetuned is a text classification model from billster45. Use it when you need a label for a piece of text. It is set up for transformers.
- Problem type: Multi-class Classification
All-time downloads
428
Public
Repo size
804 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.bin268 MB · 100%
At a glance
- Task
- Text Classification
- Library
- transformers
- Model type
- distilbert
- Access
- Public
- Created
- Apr 9, 2023
- Updated
- Apr 10, 2023
- SHA
- 385fec24
Try a prompt
Model card
From the Hugging Face model README
Model Trained Using AutoTrain
- Problem type: Multi-class Classification
Validation Metrics
| precision | recall | f1-score | support | |
|---|---|---|---|---|
| World News | 0.97 | 0.94 | 0.95 | 955 |
| Sports News | 0.99 | 0.99 | 0.99 | 938 |
| Business News | 0.93 | 0.89 | 0.91 | 991 |
| Science-Technology News | 0.88 | 0.95 | 0.91 | 916 |
| accuracy | 0.94 | 3800 | ||
| macro avg | 0.94 | 0.94 | 0.94 | 3800 |
| weighted avg | 0.94 | 0.94 | 0.94 | 3800 |
- Task
- Text Classification
- Library
- transformers
- Type
- distilbert
- Languages
- en
- Created
- Apr 9, 2023
- Updated
- Apr 10, 2023