
半监督文本分类中的对抗训练代码方法
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
本研究探讨了在半监督环境下利用对抗训练改进文本分类精度的方法,并提供了相应的实现代码。
半监督文本分类的对抗训练方法规范了代码重现过程。为了设置环境,请安装所需的依赖项。您可以使用提供的命令轻松下载预训练模型。
```
wget http://sato-motoki.com/research/vat/imdb_pretrained_lm.model
```
结果如下:
- 结果模型错误率基线:7.39
- 基准(我们的代码): 6.62
- 对抗性: 6.21
- 对抗训练(我们的代码) : 6.35
- 虚拟对抗训练 : 6.40
- 虚拟对抗训练(我们的代码) :5.82
运行预训练模型时,使用以下命令:
```
python -u pretrain.py -g 0 --layer
```
全部评论 (0)
还没有任何评论哟~


