Skip to content
master
Switch branches/tags
Go to file
Code

Latest commit

 

Git stats

Files

Permalink
Failed to load latest commit information.
Type
Name
Latest commit message
Commit time
 
 
src
 
 
 
 

README.md

FM-GAN

This repository is the implementation for paper Adversarial Text Generation via Feature-Mover’s Distance

Requirement:

Python 2.7, Tensorflow 1.8.0

Run

  • Run: python autoencoder.py for MLE pre-train
  • Run: python text_GAN.py for adversarial training
  • Options: options can be made by changing option class.

Note:

model.py: sinkhorn divergence edition

model2.py: IPOT edition

Dataset:

MSCOCO dataset and WMT news dataset can be downloaded from link HERE: Download link

Evaluation:

Use convert_new.py to convert the indexed files to sentences. Then use selfbleu.py and testbleu.py to evaluate the results.

Note that it is just a rough edition, you have to change the file names manually in the code, we will update this ASAP.

TODO:

  1. Clean the code, make it easy to understand.
  2. Don't need to manually change the file names in the code.
  3. etc.

Citation

Please cite our paper if it helps with your research

@inproceedings{chen2018adversarial,
  title={Adversarial Text Generation via Feature-Mover's Distance},
  author={Chen, Liqun and Dai, Shuyang and Tao, Chenyang and Shen, Dinghan and Gan, Zhe and Zhang, Haichao and Zhang, Yizhe and Carin, Lawrence},
  Booktitle={NIPS},
  year={2018}
}

For any question or suggestions, feel free to contact my email.

About

Tensorflow implementation for paper "Adversarial Text Generation via Feature-Mover’s Distance"

Resources

Releases

No releases published

Packages

No packages published

Languages