How can i get the topic distribution of a document?

adjidieng / ETM

Topic Modeling in Embedding Spaces

MIT License

549 stars 128 forks source link

Closed ZJiaBin closed 4 years ago

ZJiaBin commented 4 years ago

It would be grateful if anybody can tell me how to do that ^_^

ZJiaBin commented 4 years ago

Oh, maybe this will do that ~ torch.mm(data_batch,torch.transpose(beta,1,0))

ydennisy commented 4 years ago

@ZJiaBin did you ever figure out how to do this correctly?