generative adversarial nets for neural machine translation
This is an open source implementation of our framework to improve NMT with conditional sequence generative adversarial nets.which is described in the following papers:
Yang, Z., Chen, W., Wang, F., & Xu, B. Improving neural machine translation with conditional sequence generative adversarial nets. (NAACL 2018)
Requirements: Tensorflow 1.2.0, python 2.x
Useage:
pre-train the discriminator by: sh discriminator_pretrain.sh
pre-train the generator by: sh train.sh
generate the samples by: sh generate_sample.sh
run the gan training by: sh gan_train.sh