(forked) textGAN (Adversarial Feature Matching for Text Generation)

Theano and tensorflow implementation for

  • Adversarial Feature Matching for Text Generation, Yizhe Zhang, Zhe Gan, Kai Fan, Zhi Chen, Ricardo Henao, Lawrence Carin. ICML, 2017.
  • Generating Text via Adversarial Training. Yizhe Zhang, Zhe Gan, Lawrence Carin.  Workshop on Adversarial Training, NIPS, 2016.

Run

(Tensorflow) Run:

python textGAN.py

Prerequisite

  • Python3
  • Tensorflow:
    • Tensorflow version >= 1.2
    • cPickle

Data link

Three_corpus

Arxiv only

Pretrained parameter

http://people.duke.edu/~yz196/zip/param.zip

Evaluation

  • (Theano only) python eval_kde.py

Citation

Please cite our paper if it helps with your research.

@inproceedings{zhang2017adversarial,
  title={Adversarial Feature Matching for Text Generation},
  author={Zhang, Yizhe and Gan, Zhe and Fan, Kai and Chen, Zhi and Henao, Ricardo and Shen, Dinghan and Carin, Lawrence},
  booktitle={ICML},
  year={2017}
}