ETM
===

.. py:module:: topmost.models.basic.ETM



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



.. autoapisummary::


   topmost.models.basic.ETM.ETM





.. py:class:: ETM(vocab_size, embed_size=200, num_topics=50, en_units=800, dropout=0.0, pretrained_WE=None, train_WE=False)

   Bases: :py:obj:`torch.nn.Module`


   Topic Modeling in Embedding Spaces. TACL 2020

   Adji B. Dieng, Francisco J. R. Ruiz, David M. Blei.


   .. py:attribute:: topic_embeddings


   .. py:attribute:: encoder1


   .. py:attribute:: fc21


   .. py:attribute:: fc22


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


   .. py:method:: encode(x)


   .. py:method:: get_theta(x)


   .. py:method:: get_beta()


   .. py:method:: forward(x, avg_loss=True)


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