basic_trainer#
Module Contents#
- 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)#