对比学习+Prompt+FGSM实现文本分类

本文主要是基于Robert的文本分类任务,在此基础上考虑融合对比学习、Prompt和对抗训练来提升模型的文本分类能力,我本地有SST-2数据集的train.txt、test.txt、dev.txt三个文件,每个文件包含文本内容和标签两列,基于pytorch实现。

目录

一、加载相关库

二、加载数据集

三、对比学习的数据增强

你可能感兴趣的:(文本分类,学习,pytorch,人工智能)