LeakGAN Discriminator

class textbox.module.Discriminator.LeakGANDiscriminator.LeakGANDiscriminator(config, dataset)[source]

Bases: UnconditionalGenerator

CNN based discriminator for leakgan extracting feature of current sentence

calculate_loss(real_data, fake_data)[source]

Calculate discriminator loss and acc

forward(data)[source]

Get current sentence feature by CNN

get_feature(inp)[source]

Get feature vector of given sentences

Parameters

inp – batch_size * max_seq_len

Returns

batch_size * feature_dim

highway(data)[source]
training: bool