trainers#

Package Contents#

`BasicTrainer`
`LDAGensimTrainer`
`LDASklearnTrainer`
`NMFGensimTrainer`
`NMFSklearnTrainer`
`CrosslingualTrainer`
`DynamicTrainer`
`DTMTrainer`
`HierarchicalTrainer`
`HDPGensimTrainer`

class BasicTrainer(model, epochs=200, learning_rate=0.002, batch_size=200, lr_scheduler=None, lr_step_size=125, log_interval=5)#

make_optimizer()#

make_lr_scheduler(optimizer)#

fit_transform(dataset_handler, num_top_words=15, verbose=False)#

train(dataset_handler, verbose=False)#

test(input_data)#

export_beta()#

export_top_words(vocab, num_top_words=15)#

export_theta(dataset_handler)#

class LDAGensimTrainer(dataset, num_topics=50, max_iter=1, alpha='symmetric', eta=None)#

train()#

test(bow)#

export_beta()#

export_top_words(num_top=15)#

export_theta()#

class LDASklearnTrainer(model, dataset)#

train()#

test(bow)#

export_beta()#

export_top_words(num_top=15)#

export_theta()#

class NMFGensimTrainer(dataset_handler, vocab_size, num_topics=50, max_iter=1)#

train()#

test(bow)#

export_beta()#

export_top_words(num_top=15)#

export_theta()#

class NMFSklearnTrainer(model, dataset_handler)#

train()#

test(bow)#

export_beta()#

export_top_words(num_top=15)#

export_theta()#

class CrosslingualTrainer(model, epochs=500, learning_rate=0.002, batch_size=200, lr_scheduler=None, lr_step_size=125, log_interval=5)#

make_optimizer()#

make_lr_scheduler(optimizer)#

fit_transform(dataset_handler, num_top_words=15, verbose=False)#

train(dataset_handler, verbose=False)#

get_theta(bow, lang)#

test(bow_en, bow_cn)#

export_beta()#

export_top_words(vocab_en, vocab_cn, num_top_words=15)#

export_theta(dataset_handler)#

class DynamicTrainer(model, epochs=200, learning_rate=0.002, batch_size=200, lr_scheduler=None, lr_step_size=125, log_interval=5)#

make_optimizer()#

make_lr_scheduler(optimizer)#

fit_transform(dataset_handler, num_top_words=15, verbose=False)#

train(dataset_handler, verbose=False)#

test(bow, times)#

export_beta()#

export_top_words(vocab, num_top_words=15)#

export_theta(dataset_handler)#

class DTMTrainer(dataset_handler, num_topics=50, alphas=0.01, chain_variance=0.005, passes=10, lda_inference_max_iter=25, em_min_iter=6, em_max_iter=20)#

train()#

test(bow)#

get_theta()#

export_beta()#

export_top_words(num_top=15)#

export_theta()#

class HierarchicalTrainer(model, epochs=200, learning_rate=0.002, batch_size=200, lr_scheduler=None, lr_step_size=125, log_interval=5)#

make_optimizer()#

make_lr_scheduler(optimizer)#

fit_transform(dataset_handler, num_top_words=15, verbose=False)#

train(dataset_handler, verbose=False)#

test(bow)#

export_phi()#

export_beta()#

export_top_words(vocab, num_top_words=15, annotation=False)#

export_theta(dataset_handler)#

class HDPGensimTrainer(dataset, max_chunks=None, max_time=None, chunksize=256, kappa=1.0, tau=64.0, K=15, T=150, alpha=1, gamma=1, eta=0.01, scale=1.0, var_converge=0.0001)#

train()#

test(bow)#

export_beta()#

export_top_words(num_top_words=15)#

export_theta()#