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