Downloads · 30 days
6
10% of all-time downloads
heezuss/news-distilbert-full
news-distilbert-full is a machine learning model from heezuss. Use it for the machine learning task on the model card, and read the license before you ship it in a product.
Fine-tuned transformer model for classifying news articles into four categories: World, Sports, Business, and Sci/Tech.
Downloads · 30 days
6
10% of all-time downloads
All-time downloads
63
Public
Parameters
67M
268 MB on disk
Likes
0
Public
Click a slice to open those files.
.safetensors268 MB · 100%
From the Hugging Face model README
Fine-tuned transformer model for classifying news articles into four categories: World, Sports, Business, and Sci/Tech.
Dataset: AG News
Fine-tuning: Supervised (Full fine-tuning or LoRA-based fine-tuning)
Maximum sequence length: 128
This model was trained and evaluated as part of an academic project on applied natural language processing and transformer-based text classification.