N

News Category Classifier

由ranudee開發
基於DistilBERT架構的輕量級文本分類模型,用於將英文新聞分類為國際、體育、商業和科技四大類別
下載量 526
發布時間 : 11/20/2024

模型概述

該模型是在AG新聞數據集上微調的DistilBERT變體,專門用於英文新聞文章的四分類任務。作為BERT的輕量化版本,它在保持較高準確率的同時提升了推理速度。

模型特點

輕量高效
採用DistilBERT架構,相比原始BERT模型體積減小40%的同時保留97%的性能
多分類能力
可準確識別新聞所屬的四大類別(國際/體育/商業/科技)
即時推理
優化後的架構適合需要快速響應的生產環境部署

模型能力

英文文本分類
新聞內容分析
即時文本處理

使用案例

媒體分析
新聞自動分類系統
為新聞聚合平臺自動打標籤
實現90%+的分類準確率(基於推斷數據)
商業智能
行業趨勢監控
從新聞流中識別商業科技類內容
AIbase
智啟未來,您的人工智能解決方案智庫
© 2025AIbase