This is the official implementation of the paper Learning to Encode Text as Human-Readable Summaries using Generative Adversarial Networks. If you use this code or our results in your research, we'd appreciate you cite our paper as following:

@article{Wang2018Summary_GAN,
  title={Learning to Encode Text as Human-Readable Summaries using Generative Adversarial Networks},
  author={Yau-Shian Wang and Hung-Yi Lee},
  journal={arXiv preprint arXiv:1810.02851},
  year={2018}
}

• python3
• 1.0 <= tensorflow < 2.0

In this implementation, I use the GAN training method proposed by ScratchGAN for adversarial training. The performance is more robust and slightly better than original paper.

Download data from Sent-Summary. Then, move downloaded "train.article.txt" and "train.title.txt" to "giga_word" directory in this repository.

python3 make_pretrain.py

python3 main.py -pretrain -model_dir [model_path] -num_steps 20000

Pretraining generator is required.

python3 main.py -train -model_dir [model_path]-num_steps 7000

The model_path should be same as the pretrained model path. The default setting can reproduce the results in the paper.

python3 main.py -test -model_dir [model_path] -test_input [input_path] -result_path [result_path]