ETM#
Module Contents#
Topic Modeling in Embedding Spaces. TACL 2020 |
- class ETM(vocab_size, embed_size=200, num_topics=50, en_units=800, dropout=0.0, pretrained_WE=None, train_WE=False)#
Bases:
torch.nn.ModuleTopic Modeling in Embedding Spaces. TACL 2020
Adji B. Dieng, Francisco J. R. Ruiz, David M. Blei.
- reparameterize(mu, logvar)#
- encode(x)#
- get_theta(x)#
- get_beta()#
- forward(x, avg_loss=True)#
- loss_function(x, recon_x, mu, logvar, avg_loss=True)#