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.

sinkhorn_alpha

OT_max_iter = 5000

weight_loss_ECR

stopThr = 0.005

epsilon = 1e-16

forward(M)