Contents:

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.

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)#