A

Ag News Distilbert

由 huggingfacecoin 开发
基于distilbert-base-uncased微调的文本分类模型,在AG News数据集上训练,准确率达89.06%
下载量 31
发布时间 : 5/17/2025
模型介绍
内容详情
替代品

模型简介

该模型是DistilBERT的微调版本,专门用于新闻分类任务,能够将新闻文本分类到预定义的类别中

模型特点

高效轻量
基于DistilBERT架构,在保持较高性能的同时显著减小模型体积
高准确率
在AG News数据集上达到89.06%的分类准确率
快速推理
相比完整BERT模型,推理速度更快

模型能力

新闻文本分类
英文文本理解
短文本分类

使用案例

新闻媒体
新闻自动分类
将新闻文章自动分类到预定义的类别中
准确率89.06%
内容管理
内容自动标记
为网站或应用中的新闻内容自动添加分类标签