基于DistilBERT架构的轻量级文本分类模型,通过TextAttack框架在AG News数据集上微调,适用于新闻分类任务。
下载量 544
发布时间 : 3/2/2022
模型简介
该模型是DistilBERT的微调版本,专门用于序列分类任务,特别是在新闻分类场景中表现优异。
模型特点
高效轻量
基于DistilBERT架构,在保持BERT 95%性能的同时体积减小40%
高准确率
在AG News验证集上达到94.79%的准确率
对抗训练
通过TextAttack框架进行训练,可能具备更好的对抗鲁棒性
模型能力
英语文本分类
新闻类别预测
短文本分析
使用案例
新闻媒体
新闻自动分类
自动将新闻文章分类到预定义的类别中
验证准确率94.79%
内容管理
内容审核辅助
帮助识别和分类用户生成的内容
精选推荐AI模型
Llama 3 Typhoon V1.5x 8b Instruct
专为泰语设计的80亿参数指令模型,性能媲美GPT-3.5-turbo,优化了应用场景、检索增强生成、受限生成和推理任务
大型语言模型
Transformers

支持多种语言
L
scb10x
3,269
16
Cadet Tiny
Openrail
Cadet-Tiny是一个基于SODA数据集训练的超小型对话模型,专为边缘设备推理设计,体积仅为Cosmo-3B模型的2%左右。
对话系统
Transformers

英语
C
ToddGoldfarb
2,691
6
Roberta Base Chinese Extractive Qa
基于RoBERTa架构的中文抽取式问答模型,适用于从给定文本中提取答案的任务。
问答系统
中文
R
uer
2,694
98
AIbase是一个专注于MCP服务的平台,为AI开发者提供高质量的模型上下文协议服务,助力AI应用开发。
简体中文