NMF_trainer

Module Contents

NMFGensimTrainer
NMFSklearnTrainer

class NMFGensimTrainer(dataset, num_topics=50, num_top_words=15, max_iter=1)

dataset

num_topics = 50

num_top_words = 15

vocab_size

max_iter = 1

train()

test(bow)

get_beta()

get_top_words(num_top_words=None)

export_theta()

class NMFSklearnTrainer(model, dataset, num_top_words=15)

model

dataset

num_top_words = 15

train()

test(bow)

get_beta()

get_top_words(num_top_words=None)

export_theta()