news-topic-classification-with-bert

作者 wesleyacheng · 下载 76 · 点赞 1 · 更新 2023-11-02

BREAKING NEWS: We are making a news topic classifier!

First posted on my Kaggle.

Here I did transfer learning on BERT with the AG News Dataset to create a multi-class news topic classifier to classify news as either Business, Sports, Sci/Tech, or World.