search

CR-Gjx / LeakGAN

The codes of paper "Long Text Generation via Adversarial Training with Leaked Information" on AAAI 2018. Text generation using GAN and Hierarchical Reinforcement Learning.
https://arxiv.org/abs/1709.08624
576 stars 180 forks source link

Variable start_token #12

Open Crista23 opened 6 years ago

Crista23 commented 6 years ago

In Main.py, line 22 (https://github.com/CR-Gjx/LeakGAN/blob/master/Image%20COCO/Main.py#L22 ) START_TOKEN is set to 0. I wonder is it possible to use LeakGAN to generate text samples in batch where each sample starts with a different START_TOKEN?

CR-Gjx commented 6 years ago

I think it's another kind of conditional text generation, likes translation(given text, generate translation), the start token or start hidden_state like condition. This is my guess, welcome to discuss.