NMTM¶

Module Contents¶

Learning Multilingual Topics with Neural Variational Inference. NLPCC 2020.

class NMTM(Map_en2cn, Map_cn2en, vocab_size_en, vocab_size_cn, num_topics=50, en_units=200, dropout=0.0, lam=0.8)¶

Bases: torch.nn.Module

Learning Multilingual Topics with Neural Variational Inference. NLPCC 2020.

Xiaobao Wu, Chunping Li, Yan Zhu, Yishu Miao.

num_topics = 50¶

lam = 0.8¶

Map_en2cn¶

Map_cn2en¶

a¶

mu2¶

var2¶

decoder_bn_en¶

decoder_bn_cn¶

fc11_en¶

fc11_cn¶

fc12¶

fc21¶

fc22¶

fc1_drop¶

z_drop¶

mean_bn¶

logvar_bn¶

phi_en¶

phi_cn¶

reparameterize(mu, logvar)¶

encode(x, lang)¶

get_theta(x, lang)¶

get_beta()¶

decode(theta, lang)¶

forward(x_en, x_cn)¶

loss_function(recon_x, x, mu, logvar)¶