Contents:

DecTM#

Module Contents#

Discovering Topics in Long-tailed Corpora with Causal Intervention. ACL 2021 findings.

class DecTM(vocab_size, num_topics=50, en_units=200, dropout=0.4)#

Bases: torch.nn.Module

Discovering Topics in Long-tailed Corpora with Causal Intervention. ACL 2021 findings.

Xiaobao Wu, Chunping Li, Yishu Miao.

get_beta()#

get_theta(x)#

reparameterize(mu, logvar)#

encode(x)#

decode(theta)#

forward(x)#

loss_function(x, recon_x, mu, logvar)#