LantaoYu / SeqGAN

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

Modify get_reward to suit various length sequence #42

Closed TobiasLee closed 6 years ago

TobiasLee commented 6 years ago

change for given_num in range(1, 20) : to for given_num in range(1, self.sequence_length ):

and reward[19] += yred to reward[self.sequence_length - 1 ] += yred