ECR#

Module Contents#

ECR

Effective Neural Topic Modeling with Embedding Clustering Regularization. ICML 2023

class ECR(weight_loss_ECR, sinkhorn_alpha, OT_max_iter=5000, stopThr=0.005)#

Bases: torch.nn.Module

Effective Neural Topic Modeling with Embedding Clustering Regularization. ICML 2023

Xiaobao Wu, Xinshuai Dong, Thong Thanh Nguyen, Anh Tuan Luu.

forward(M)#