TAMI#

Module Contents#

TAMI

InfoCTM: A Mutual Information Maximization Perspective of Cross-lingual Topic Modeling. AAAI 2023
class TAMI(temperature, weight_MI, pos_threshold, trans_e2c, pretrain_word_embeddings_en, pretrain_word_embeddings_cn)#

Bases: torch.nn.Module

InfoCTM: A Mutual Information Maximization Perspective of Cross-lingual Topic Modeling. AAAI 2023

Xiaobao Wu, Xinshuai Dong, Thong Nguyen, Chaoqun Liu, Liangming Pan, Anh Tuan Luu

build_CVL_mask(embeddings)#
translation_mask(mask, trans_dict_matrix)#
compute_pos_neg(pretrain_word_embeddings_en, pretrain_word_embeddings_cn, trans_e2c, trans_c2e)#
MutualInfo(anchor_feature, contrast_feature, mask, neg_mask)#
forward(fea_en, fea_cn)#