OCTIS: Comparing Topic Models is Simple! A python package to optimize and evaluate topic models (accepted at EACL2021 demo track)
705
stars
98
forks
source link
passing NeuralLDA.train_model(top_words=10) to AVITM_model.train_model(top_words=10) #103
Open
hosseinfani opened 1 year ago
using top_words for topics' top-k words in neurallda >> AVITM >> AVITM_model >> train_model() >> get_info()
https://github.com/MIND-Lab/OCTIS/blob/068565507cd08bf88cb314fa953b08812e8cacef/octis/models/pytorchavitm/AVITM.py#L53
Also, adding verbose