search

williamSYSU / TextGAN-PyTorch

TextGAN is a PyTorch framework for Generative Adversarial Networks (GANs) based text generation models.
MIT License
878 stars 207 forks source link

Is the sentigan for multiclass data generation? #26

Open rakaiv5 opened 4 years ago

rakaiv5 commented 4 years ago

The figure of the sentiGAN show that the discriminator of sentiGAN is a multi-class classifier, but in your code, I think it is a binary classifier. So the difference between sentiGAN and seqGAN is just the penalty based objective?

williamSYSU commented 4 years ago

The discriminator of SentiGAN is a multi-class classifier, which outputs a k + 1 dimension vector, where k is the number of categories. Another difference between SentiGAN and SeqGAN is the penalty based objective.