def __init__(self, corpus, config, action2name):
super(GanAgent_AutoEncoder_State,
self).__init__(corpus, config, action2name)
self.discriminator = gan_model_sat.WoZDiscriminator(config)
self.generator = gan_model_sat.WoZGenerator_StateVae(config)
self.autoencoder_in_size = config.state_out_size
self.vae = gan_model_vae.AutoEncoder(config)
def __init__(self, corpus, config, action2name):
super(GanAgent_SAT_WoZ, self).__init__(corpus, config, action2name)
self.discriminator =gan_model_sat.WoZDiscriminator(config)
self.generator = gan_model_sat.WoZGenerator(config)
self.state_out_size = config.state_out_size
self.noise_factor = 0.3