Returns news category given text.
See https://www.kaggle.com/code/dima806/news-category-classification-distilbert for more details.
Classification report:
precision recall f1-score support
ARTS 0.4985 0.5629 0.5288 302
ARTS & CULTURE 0.5061 0.4627 0.4834 268
BLACK VOICES 0.5903 0.4776 0.5280 917
BUSINESS 0.6136 0.5659 0.5888 1198
COLLEGE 0.5043 0.5066 0.5054 229
COMEDY 0.5990 0.5630 0.5804 1080
CRIME 0.6365 0.6615 0.6488 712
CULTURE & ARTS 0.7133 0.4744 0.5698 215
DIVORCE 0.8498 0.8015 0.8249 685
EDUCATION 0.5000 0.5025 0.5012 203
ENTERTAINMENT 0.7383 0.8146 0.7745 3473
ENVIRONMENT 0.5490 0.5433 0.5461 289
FIFTY 0.6107 0.3250 0.4242 280
FOOD & DRINK 0.7514 0.8320 0.7897 1268
GOOD NEWS 0.4676 0.2321 0.3103 280
GREEN 0.4685 0.5401 0.5018 524
HEALTHY LIVING 0.5669 0.4712 0.5147 1339
HOME & LIVING 0.8267 0.8113 0.8189 864
IMPACT 0.5000 0.3702 0.4254 697
LATINO VOICES 0.6066 0.5664 0.5858 226
MEDIA 0.6136 0.5688 0.5903 589
MONEY 0.6193 0.5840 0.6012 351
PARENTING 0.6711 0.7673 0.7160 1758
PARENTS 0.5094 0.4475 0.4764 791
POLITICS 0.8154 0.8365 0.8258 7120
QUEER VOICES 0.7949 0.7392 0.7660 1269
RELIGION 0.6681 0.6097 0.6376 515
SCIENCE 0.6370 0.6327 0.6348 441
SPORTS 0.7628 0.8079 0.7847 1015
STYLE 0.6343 0.6231 0.6286 451
STYLE & BEAUTY 0.8656 0.8894 0.8773 1962
TASTE 0.4701 0.4320 0.4502 419
TECH 0.6188 0.5952 0.6068 420
THE WORLDPOST 0.5786 0.5825 0.5806 733
TRAVEL 0.8501 0.8596 0.8548 1980
U.S. NEWS 0.4256 0.3018 0.3532 275
WEDDINGS 0.8320 0.8810 0.8558 731
WEIRD NEWS 0.5030 0.4559 0.4783 555
WELLNESS 0.7272 0.8459 0.7821 3589
WOMEN 0.4841 0.4062 0.4417 714
WORLD NEWS 0.4936 0.4682 0.4806 660
WORLDPOST 0.6840 0.6376 0.6600 516
accuracy 0.7073 41903
macro avg 0.6275 0.5966 0.6080 41903
weighted avg 0.7007 0.7073 0.7017 41903
- Downloads last month
- 14,206
This model does not have enough activity to be deployed to Inference API (serverless) yet. Increase its social
visibility and check back later, or deploy to Inference Endpoints (dedicated)
instead.
Model tree for dima806/news-category-classifier-distilbert
Base model
google-bert/bert-base-uncased