adjidieng / ETM

Topic Modeling in Embedding Spaces
MIT License
538 stars 126 forks source link

How can i get the topic distribution of a document? #11

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 3 years ago

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