# news-category-classification-distilbert

**Repository Path**: modelee/news-category-classification-distilbert

## Basic Information

- **Project Name**: news-category-classification-distilbert
- **Description**: No description available
- **Primary Language**: Unknown
- **License**: Not specified
- **Default Branch**: main
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 4
- **Forks**: 0
- **Created**: 2023-05-24
- **Last Updated**: 2025-07-29

## Categories & Tags

**Categories**: llm

**Tags**: None

## README

---
license: openrail
language:
- en
---
# distilbert-base-news-category-classification
This repository provides a distilbert model train on 210k news headlines from 2012 to 2022 from [HuffPost](https://www.huffpost.com/). 
The model was trained by [Yue](https://github.com/yueeeeeee87)

The training data can be found at the following Kaggle URL.
[https://www.kaggle.com/datasets/rmisra/news-category-dataset](https://www.kaggle.com/datasets/rmisra/news-category-dataset)

Model and Project details:
https://yueh-huan.com/posts/identify-news-category-based-on-news-headlines/