If you have any question or suggestion about this implementation, please do not hesitate to contact me. As a comparison to the NTM, an out-of-box LDA script is also provided, which is based on the gensim library. Datasets of short news ( cnews10k), dialogue utterances ( zhddline) and conversation ( zhdd), are presented for evaluation purpose, all of which are in Chinese. Configuration of the models will not exactly the same as those proposed in the papers, and the hyper-parameters are not carefully finetuned, but I have chosen to get the core ideas covered.Įmpirically, NTM is superior to classical statistical topic models ,especially on short texts. The aim of this project is to provide a practical and working example for neural topic models to facilitate the research of related fields. PyTorch implementations of Neural Topic Model varieties proposed in recent years, including NVDM-GSM, WTM-MMD (W-LDA), WTM-GMM, ETM, BATM ,and GMNTM.
0 Comments
Leave a Reply. |
AuthorWrite something about yourself. No need to be fancy, just an overview. ArchivesCategories |