No parameter for topk words per topic
ERijck opened this issue · 2 comments
ERijck commented
- OCTIS version: 1.83
- Python version: 3.8
- Operating System: Windows
Description
When training a model, I want to adjust the number of words per topic. However, I did not see this option in the source code of various models. Can you add this option for the different models?
silviatti commented
Hi,
the parameter top_words in the train_model function controls what you want. For example,
model = LDA(num_topics=25)
model_output = model.train_model(dataset, top_words=20)I noticed that this wasn't present in the implementation of NMF and I fixed it. It will be part of the next release. If you notice that it's missing in other models that you want to use, please let me know.
Thanks,
Silvia