Robert+SimCLR+FGSM实现文本分类

Robert文本分类基础上,我用的是GLUE的SST-2数据集,包含train.txt、test.txt、dev.txt三个文件,每个文件包含内容和标签两列。用SimCLR思想结合对抗训练的思想提升模型文本分类的准确率,我用Pytorch实现,代码逐行注释。

目录

二、加载数据集

三、定义模型

四、定义训练函数

五、定义测试函数

你可能感兴趣的:(文本分类,分类,深度学习,机器学习)