UWE¶

Module Contents¶

class UWE(ETC, num_times, temperature, weight_UWE, neg_topk)¶

Bases: torch.nn.Module

ETC¶

weight_UWE¶

num_times¶

temperature¶

neg_topk¶

forward(time_wordcount, beta, topic_embeddings, word_embeddings)¶

get_topk_indices(beta)¶