hierarchical_trainer#
Module Contents#
|
- to_nparray(tensor_list)#
- 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)#