ECR
===

.. py:module:: topmost.models.basic.ECRTM.ECR



Module Contents
---------------



.. autoapisummary::


   topmost.models.basic.ECRTM.ECR.ECR





.. py:class:: ECR(weight_loss_ECR, sinkhorn_alpha, OT_max_iter=5000, stopThr=0.005)

   Bases: :py:obj:`torch.nn.Module`


   Effective Neural Topic Modeling with Embedding Clustering Regularization. ICML 2023

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


   .. py:attribute:: sinkhorn_alpha


   .. py:attribute:: OT_max_iter
      :value: 5000



   .. py:attribute:: weight_loss_ECR


   .. py:attribute:: stopThr
      :value: 0.005



   .. py:attribute:: epsilon
      :value: 1e-16



   .. py:method:: forward(M)