news-category-classifier-distilbert

dima806/news-category-classifier-distilbert

news-category-classifier-distilbert is a text classification model from dima806. Use it when you need a label for a piece of text. It is set up for transformers. The card lists the license as apache-2.0.

Downloads · 30 days

399

0% of all-time downloads

All-time downloads

211K

Public

Parameters

110M

1.8 GB on disk

Likes

Public

Hugging Face

Repo makeup

Click a slice to open those files.

.pt876 MB · 40%

At a glance

Task: Text Classification
Library: transformers
License: apache-2.0
Model type: bert
Access: Public
Created: Sep 26, 2023
Updated: Oct 19, 2024
SHA: a5d6e7f9

Try a prompt

Base models

google-bert/bert-base-uncased

Model card

From the Hugging Face model README

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

Task: Text Classification
Library: transformers
Type: bert
License: apache-2.0
Created: Sep 26, 2023
Updated: Oct 19, 2024