search

adjidieng / ETM

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

How to modify the code to number of topics other than 50? #25

Open shivanibhoite opened 3 years ago

shivanibhoite commented 3 years ago

What changes can we make to the code where we can change the number of topics from 50 to any other number? For my data I need to have only 5-6 topics.

I did change in number of topics in the first command that creates the embedding vectors, I was able to do it, but I am facing the following issue.

Screen Shot 2020-12-12 at 15 38 42
jfcann commented 3 years ago

Hi shivanibhoite, you need to specify the --num_topics argument when running main.py with --mode eval. For your case that would be --num_topics 5.