D

Distilbert Base Uncased Ag News

由 textattack 开发
基于DistilBERT架构的轻量级文本分类模型,通过TextAttack框架在AG News数据集上微调,适用于新闻分类任务。
下载量 544
发布时间 : 3/2/2022
模型介绍
内容详情
替代品

模型简介

该模型是DistilBERT的微调版本,专门用于序列分类任务,特别是在新闻分类场景中表现优异。

模型特点

高效轻量
基于DistilBERT架构,在保持BERT 95%性能的同时体积减小40%
高准确率
在AG News验证集上达到94.79%的准确率
对抗训练
通过TextAttack框架进行训练,可能具备更好的对抗鲁棒性

模型能力

英语文本分类
新闻类别预测
短文本分析

使用案例

新闻媒体
新闻自动分类
自动将新闻文章分类到预定义的类别中
验证准确率94.79%
内容管理
内容审核辅助
帮助识别和分类用户生成的内容