D

Distilbert Imdb

由 lvwerra 开发
基于DistilBERT-base-uncased在IMDB影评数据集上微调的文本分类模型,用于情感分析任务
下载量 14.18k
发布时间 : 3/2/2022
模型介绍
内容详情
替代品

模型简介

该模型是DistilBERT的轻量级版本,专门针对IMDB电影评论数据集进行微调,用于二分类情感分析(正面/负面评论)。在评估集上达到92.8%的准确率。

模型特点

高效轻量
基于DistilBERT架构,比标准BERT模型体积小40%,速度快60%,同时保留97%的性能
高准确率
在IMDB测试集上达到92.8%的分类准确率
简单易用
通过Hugging Face Transformers库可快速部署使用

模型能力

文本分类
情感分析
英语文本处理

使用案例

情感分析
影评情感分类
自动判断电影评论的情感倾向(正面/负面)
准确率92.8%
产品评论分析
分析电商平台上的用户评价情感倾向