LeakGAN Discriminator¶
- class textbox.module.Discriminator.LeakGANDiscriminator.LeakGANDiscriminator(config, dataset)[source]¶
Bases:
UnconditionalGenerator
CNN based discriminator for leakgan extracting feature of current sentence
- get_feature(inp)[source]¶
Get feature vector of given sentences
- Parameters
inp – batch_size * max_seq_len
- Returns
batch_size * feature_dim
- training: bool¶