news-category-classifier-distilbert開源新聞分類模型

首頁

News Category Classifier Distilbert

由dima806開發

基於BERT-base-uncased的新聞分類模型，能夠對多種新聞類別進行準確分類。

文本分類

Transformers

開源協議:Apache-2.0 #BERT優化 #多類別分類 #高準確率

下載量 342

發布時間 : 9/26/2023

模型概述

該模型使用BERT-base-uncased作為基礎架構，經過微調後用於新聞分類任務，支持多種新聞類別的識別。

模型特點

高準確率

在多個新聞類別上表現出色，整體準確率達到70.73%。

多類別支持

支持超過30種新聞類別的分類，涵蓋政治、娛樂、體育等多個領域。

基於BERT架構

利用BERT的強大語義理解能力，提供高質量的文本分類。

模型能力

新聞分類

文本理解

多類別識別

使用案例

新聞聚合

新聞自動分類

將新聞文章自動分類到預定義的類別中，如政治、娛樂、體育等。

準確率70.73%，F1值60.80%

內容推薦

個性化新聞推薦

根據用戶閱讀歷史，推薦相關類別的新聞內容。

提高用戶粘性和閱讀時長

🚀 新聞分類模型

本項目提供了一個新聞分類模型，可根據輸入的文本返回對應的新聞類別。該模型具有一定的準確性和實用性，能為新聞內容的分類提供有效的支持。

🚀 快速開始

該模型可根據輸入的文本返回新聞類別。更多詳細信息請參考這裡。

📄 許可證

本項目採用 Apache-2.0 許可證。

💻 使用示例

以下是該模型的分類報告：

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