NMTM
====

.. py:module:: topmost.models.crosslingual.NMTM



Module Contents
---------------



.. autoapisummary::


   topmost.models.crosslingual.NMTM.NMTM





.. py: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: :py:obj:`torch.nn.Module`


   Learning Multilingual Topics with Neural Variational Inference. NLPCC 2020.

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


   .. py:attribute:: num_topics
      :value: 50



   .. py:attribute:: lam
      :value: 0.8



   .. py:attribute:: Map_en2cn


   .. py:attribute:: Map_cn2en


   .. py:attribute:: a


   .. py:attribute:: mu2


   .. py:attribute:: var2


   .. py:attribute:: decoder_bn_en


   .. py:attribute:: decoder_bn_cn


   .. py:attribute:: fc11_en


   .. py:attribute:: fc11_cn


   .. py:attribute:: fc12


   .. py:attribute:: fc21


   .. py:attribute:: fc22


   .. py:attribute:: fc1_drop


   .. py:attribute:: z_drop


   .. py:attribute:: mean_bn


   .. py:attribute:: logvar_bn


   .. py:attribute:: phi_en


   .. py:attribute:: phi_cn


   .. py:method:: reparameterize(mu, logvar)


   .. py:method:: encode(x, lang)


   .. py:method:: get_theta(x, lang)


   .. py:method:: get_beta()


   .. py:method:: decode(theta, lang)


   .. py:method:: forward(x_en, x_cn)


   .. py:method:: loss_function(recon_x, x, mu, logvar)