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.Module

Topic Modeling in Embedding Spaces. TACL 2020

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

topic_embeddings¶

encoder1¶

fc21¶

fc22¶

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