Skip to content
forked from LantaoYu/SeqGAN

Implementation of Adversarial Ranking for Language Generation [ArxiV 1705.11001]

Notifications You must be signed in to change notification settings

desire2020/RankGAN

 
 

Repository files navigation

RankGAN

Requirements:

  • Tensorflow r1.6.0
  • Python 3.x
  • CUDA 9.0 (For GPU)

Introduction

Apply Generative Adversarial Nets to generating sequences of discrete tokens with optimization via replacing the discriminator with a ranker.

The previous research paper SeqGAN: Sequence Generative Adversarial Nets with Policy Gradient has been accepted at the Thirty-First AAAI Conference on Artificial Intelligence (AAAI-17).

The research paper Adversarial Ranking for Language Generation has been accepted at 31st Conference on Neural Information Processing Systems (NIPS 2017).

We reproduce example codes to repeat the synthetic data experiments with oracle evaluation mechanisms. To run the experiment with default parameters:

$ python sequence_gan.py

You can change the all the parameters in sequence_gan.py.

The experiment has two stages. In the first stage, use the positive data provided by the oracle model and Maximum Likelihood Estimation to perform supervise learning. In the second stage, use adversarial training to improve the generator.

Note: this code is based on the previous work by ofirnachum and SeqGAN .

About

Implementation of Adversarial Ranking for Language Generation [ArxiV 1705.11001]

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%