adjidieng / ETM

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

Add predictive measure to utils.py #35

Closed bui-thanh-lam closed 3 years ago

bui-thanh-lam commented 3 years ago

Hi,

I've read your great paper. I noticed that you measure log likelihood on test set as a predictive metric. But in utils.py, I found that you'd provided me just topic coherence and diversity measures.

Would you consider to add that predictive measure to utils.py?