半监督文本分类的对抗训练方法
规范
此代码重现用 。
设置环境
请安装和 。
您可以使用此轻松设置环境。
下载预训练模型
请下载预先训练的模型。
$ wget http://sato-motoki.com/research/vat/imdb_pretrained_lm.model
结果
模型
错误率
基线
7.39
基准(我们的代码)
6.62
对抗性
6.21
对抗训练(我们的代码)
6.35
虚拟对抗训练
6.40
虚拟对抗训练
5.91
虚拟对抗训练(我们的代码)
5.82
跑
预训练
$ python -u pretrain.py -g 0 --layer
1