Skip to content

passing NeuralLDA.train_model(top_words=10) to AVITM_model.train_model(top_words=10)#103

Open
hosseinfani wants to merge 9 commits intoMIND-Lab:masterfrom
fani-lab:master
Open

passing NeuralLDA.train_model(top_words=10) to AVITM_model.train_model(top_words=10)#103
hosseinfani wants to merge 9 commits intoMIND-Lab:masterfrom
fani-lab:master

Conversation

@hosseinfani
Copy link

using top_words for topics' top-k words in neurallda >> AVITM >> AVITM_model >> train_model() >> get_info()

def train_model(self, dataset, hyperparameters=None, top_words=10):

Also, adding verbose

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants