TopicDistQuant

Module Contents

TopicDistQuant

Short Text Topic Modeling with Topic Distribution Quantization and Negative Sampling Decoder. EMNLP 2020

class TopicDistQuant(num_embeddings, embedding_dim, commitment_cost=0.1)

Bases: torch.nn.Module

Short Text Topic Modeling with Topic Distribution Quantization and Negative Sampling Decoder. EMNLP 2020

Xiaobao Wu, Chunping Li, Yan Zhu, Yishu Miao

_embedding_dim

_num_embeddings

_embedding

_commitment_cost = 0.1

forward(inputs)