
**BREAKING NEWS: We are making a news topic classifier!**

First posted on my [Kaggle](https://www.kaggle.com/code/wesleyacheng/news-topic-classification-with-bert/notebook).

Here I did transfer learning on [BERT](https://huggingface.co/distilbert-base-uncased) with the [AG News Dataset](https://huggingface.co/datasets/ag_news) to create a multi-class news topic classifier to classify news as either Business, Sports, Sci/Tech, or World.
{"language": ["en"], "license": "apache-2.0", "datasets": ["ag_news"], "metrics": ["accuracy", "f1"], "pipeline_tag": "text-classification", "widget": [{"text": "Ukraine's President Volodymyr Zelensky appears to have confirmed that his country's long-awaited counter-offensive against Russia has started. \"Counter-offensive and defensive actions are taking place,\" he said.", "example_title": "World News Example"}, {"text": "The 2023 NFL draft class saw a record 12 quarterbacks selected in the first five rounds, including three inside the top five picks. The 2024 class? Expect some similarities.", "example_title": "Sports News Example"}, {"text": "Friday's jobs report does little to change the Federal Reserve's debate over holding rates steady this month. But it strengthens the likelihood that officials would pair any June pause with a stronger preference to raise rates later this year.", "example_title": "Business News Example"}, {"text": "According to one recent estimate, a crew of six would require an estimated 10,000 kilograms of food for the trip. NASA\u2014which plans to send people to Mars within 2 decades\u2014could stuff a spacecraft with prepackaged meals and launch additional supplies to the Red Planet in advance for the voyage home. But even that wouldn\u2019t completely solve the problem.", "example_title": "Science News Example"}]}
transformers pytorch safetensors distilbert text-classification en dataset:ag_news license:apache-2.0 autotrain_compatible endpoints_compatible region:us
TAGS
#transformers #pytorch #safetensors #distilbert #text-classification #en #dataset-ag_news #license-apache-2.0 #autotrain_compatible #endpoints_compatible #region-us 


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.
TAGS
#transformers #pytorch #safetensors #distilbert #text-classification #en #dataset-ag_news #license-apache-2.0 #autotrain_compatible #endpoints_compatible #region-us 
