search

lfmatosm / embedded-topic-model

A package to run embedded topic modelling with ETM. Adapted from the original at: https://github.com/adjidieng/ETM
MIT License
89 stars 9 forks source link

How could I get the topic embeddings？ #28

Open broboboo opened 12 months ago

broboboo commented 12 months ago

Hello! How could I get the topic embeddings, which is mentioned as alpha in the ETM original paper? Thanks and waiting for your response!

lfmatosm commented 11 months ago

Hi @broboboo, how are you? Thanks for your question. Currently, there's no method to retrieve the topic embeddings. They are initially defined here: https://github.com/lfmatosm/embedded-topic-model/blob/71996073d584ec38070dbd62095a021f80bcdb19/embedded_topic_model/models/model.py#L44 I'm currently on a deadline in other projects, but as soon as I can, I will update the package to address this shortcoming. I'm also open for pull requests, so feel free if its your cup of tea :)

broboboo commented 11 months ago

Thanks for your answer！ I found i can get the topic embeddings by this： topic_embeddings = etm_instance.model.alphas.weight.detach().cpu().numpy() Is this correct? Thank you！

lfmatosm commented 6 months ago

@broboboo yes, that's correct. Sorry for the delay.