irpas技术客

【PyTorch深度学习项目实战100例】—— 基于Transformer实现电影评论星级分类任务 | 第42例_咕 嘟_transformer分类任务

网络投稿 7608

前言

大家好,我是阿光。

本专栏整理了《PyTorch深度学习项目实战100例》,内包含了各种不同的深度学习项目,包含项目原理以及源码,每一个项目实例都附带有完整的代码+数据集。

正在更新中~ ?

?? 我的项目环境:

平台:Windows10 语言环境:python3.7 编译器:PyCharm PyTorch版本:1.8.1

?? 项目专栏:【PyTorch深度学习项目实战100例】

一、基于Transformer实现电影评论星级分类任务

本文主要用近几年兴起的Transformer拟合微调实现一个包含5个类别的电影评论星级文本分类任务,主要是对电影评论内容进行特征抽取,获取语义分析来实现分类任务。

训练集的规模有1000条,大概训练了100个epoch,最终的分类精度为96%。

二、数据集介绍

电影将所有电影分为22种不同的类型,对于每种类型,都会抓取2000条电影评论,评论按照评论者进行评论时的评分等级分为三类。

整个数据集的大小为2.3GB,为了简单测试demo,所以从所有数据中随机抽取了1000条。

Comment:电影评论 Star:分类标签,代表不


1.本站遵循行业规范,任何转载的稿件都会明确标注作者和来源;2.本站的原创文章,会注明原创字样,如未注明都非原创,如有侵权请联系删除!;3.作者投稿可能会经我们编辑修改或补充;4.本站不提供任何储存功能只提供收集或者投稿人的网盘链接。

标签: #transformer分类任务