大家好,我是阿光。
本专栏整理了《PyTorch深度学习项目实战100例》,内包含了各种不同的深度学习项目,包含项目原理以及源码,每一个项目实例都附带有完整的代码+数据集。
正在更新中~ ✨
🚨 我的项目环境:
💥 项目专栏:【PyTorch深度学习项目实战100例】
本文主要用近几年兴起的Transformer
拟合微调实现一个包含5个类别的电影评论
星级文本分类任务,主要是对电影评论内容进行特征抽取,获取语义分析
来实现分类任务。
训练集的规模有1000条,大概训练了100个epoch,最终的分类精度为96%。
电影将所有电影分为22种不同的类型,对于每种类型,都会抓取2000条电影评论,评论按照评论者进行评论时的评分等级分为三类。
整个数据集的大小为2.3GB,为了简单测试demo,所以从所有数据中随机抽取了1000条。