model_cvae.py 文件源码

def __init__(self,args):
        self.setArgs(args)
        super(CVAEHidden, self).__init__(
                categ_enc_b_h = L.EmbedID(self.categ_size,self.out_size),
                categ_enc_b_c = L.EmbedID(self.categ_size,self.out_size),
                categ_enc_f_h = L.EmbedID(self.categ_size,self.out_size),
                categ_enc_f_c = L.EmbedID(self.categ_size,self.out_size),
                categ_dec_h   = L.EmbedID(self.categ_size,2*self.out_size),
                categ_dec_c   = L.EmbedID(self.categ_size,2*self.out_size),
                embed = L.EmbedID(self.n_vocab,self.n_embed),
                #VAEenc
                enc_f = LSTM(self.n_layers,self.n_embed, self.out_size, dropout=self.drop_ratio),
                enc_b = LSTM(self.n_layers,self.n_embed, self.out_size, dropout=self.drop_ratio),

                le2_mu=L.Linear(4*self.out_size, self.n_latent),
                le2_ln_var=L.Linear(4*self.out_size, self.n_latent),
                #VAEdec
                ld_h = L.Linear(self.n_latent,2*self.out_size),
                ld_c = L.Linear(self.n_latent,2*self.out_size),

                dec = LSTM(self.n_layers,self.n_embed, 2*self.out_size, dropout=self.drop_ratio),
                h2w = L.Linear(2*self.out_size,self.n_vocab),
        )