tokmangan

Tokmangan

Tokmangan的代码：代币操纵生成的对抗网络，用于文本生成

这是一个层次序列生成模型，首先决定是否添加了空白，

如果添加了空白，则将其填充，否则决定如何使用给定令牌（使用，忽略等）。

指示

1。在MLE模式下运行Tokmangan

python train.py --gan_model tokmangan --mode MLE --dataset coco --unit_size 32

2。在gan模式下运行tokmangan

python train.py --gan_model tokmangan --mode GAN --dataset coco --unit_size 32

3。生成样品

• 用于评估

python generate_for_eval.py -g tokmangan -t GAN -d coco -s 32

• 要检查详细信息（您可以在此处找到以“详细信息”开头的结果文件）

python generate_for_details.py -g tokmangan -t GAN -d coco -s 32 --n_generate_per_seed 10 --gen_vd_keep_prob 0.8

• 您还可以在此项目上使用Maskgan型号。 （将脚本中的选项指定为-gan_model maskgan或-g maskgan ）

合成数据实验

• 您可以在此处找到合成的文本样本
• 评估代码放在这里

要求

• Python 3.6上的张托法版1.14版对代码进行了测试。
• 要进行评估，您需要下载外部项目-Gansfallingshort

参考

我们在以下项目中有很多帮助。

• maskgan
• Texygen
• Gansfallingshort