Closed arijitgupta42 closed 1 year ago
Hello, thanks for your PR. I have made a few changes:
I still have to fix a few things on my side - which are unrelated to your PR -, but I'll merge the PR after that.
Thanks again for contributing :)
Silvia
The
train_model
function in CTM has thetop_words
parameter but it doesn't get passed to thectm
class as it doesn't have any argument corresponding to the same. This results in CTM returning topics of length 10 regardless of the values oftop_words
in thetrain_model
function. For example the below code will return topics of length 10 even though we've set the value oftop_words
to 5.This PR adds the
top_word
parameter to thectm
class and modifies theget_topics
function to use the value oftop_words
passed by the user (default value is still 10)