distilbert-base-news-category-classification

This repository provides a distilbert model train on 210k news headlines from 2012 to 2022 from HuffPost. The model was trained by Yue

The training data can be found at the following Kaggle URL. 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/