Skip to content

Latest commit

 

History

History

scratchgan

ScratchGAN

This is the example code for the following NeurIPS 2019 paper. If you use the code here please cite this paper:

@article{DBLP:journals/corr/abs-1905-09922,
  author    = {Cyprien de Masson d'Autume and
               Mihaela Rosca and
               Jack W. Rae and
               Shakir Mohamed},
  title     = {Training language GANs from Scratch},
  journal   = {CoRR},
  volume    = {abs/1905.09922},
  year      = {2019},
  url       = {http://arxiv.org/abs/1905.09922},
  archivePrefix = {arXiv},
  eprint    = {1905.09922},
  timestamp = {Wed, 29 May 2019 11:27:50 +0200},
  biburl    = {https://dblp.org/rec/bib/journals/corr/abs-1905-09922},
  bibsource = {dblp computer science bibliography, https://dblp.org}
}

Contents

The code contains:

  • generators.py: implementation of the generator.
  • discriminator_nets.py: implementation of the discriminator.
  • eval_metrics.py: implementation of the FED metric.
  • losses.py: implementation of the RL loss for the generator.
  • reader.py: data reader / tokenizer.
  • experiment.py: main training script.

The data contains:

  • {train,valid,test}.json: the EMNLP2017 News dataset.
  • glove_emnlp2017.txt: the relevant subset of GloVe embeddings.

Running

./scratchgan/run.sh