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.ModuleDiscovering Topics in Long-tailed Corpora with Causal Intervention. ACL 2021 findings.
Xiaobao Wu, Chunping Li, Yishu Miao.
- num_topics = 50¶
- a¶
- mu2¶
- var2¶
- fc11¶
- fc12¶
- fc21¶
- fc22¶
- mean_bn¶
- logvar_bn¶
- decoder_bn¶
- fc1_drop¶
- theta_drop¶
- beta¶
- get_beta()¶
- get_theta(x)¶
- reparameterize(mu, logvar)¶
- encode(x)¶
- decode(theta)¶
- forward(x)¶
- loss_function(x, recon_x, mu, logvar)¶