search

LantaoYu / SeqGAN

Implementation of Sequence Generative Adversarial Nets with Policy Gradient
2.08k stars 711 forks source link

Training on custom dataset #54

Closed WorldHellow closed 4 years ago

WorldHellow commented 5 years ago

I want to train AOL Search query logs on Seq-GAN to generate queries. Is there an implementation of Seq-GAN that uses Custom Dataset?

bhushan23 commented 5 years ago

@WorldHellow we have extended Seq-GAN implementation in PyTorch which allows loading custom data. here, https://github.com/bhushan23/Transformer-SeqGAN-PyTorch