basic_trainer ============= .. py:module:: topmost.trainers.basic.basic_trainer Module Contents --------------- .. autoapisummary:: topmost.trainers.basic.basic_trainer.BasicTrainer .. autoapisummary:: topmost.trainers.basic.basic_trainer.logger .. py:data:: logger .. py:class:: BasicTrainer(model, dataset, num_top_words=15, epochs=200, learning_rate=0.002, batch_size=200, lr_scheduler=None, lr_step_size=125, log_interval=5, verbose=False) .. py:attribute:: model .. py:attribute:: dataset .. py:attribute:: num_top_words :value: 15 .. py:attribute:: epochs :value: 200 .. py:attribute:: learning_rate :value: 0.002 .. py:attribute:: batch_size :value: 200 .. py:attribute:: lr_scheduler :value: None .. py:attribute:: lr_step_size :value: 125 .. py:attribute:: log_interval :value: 5 .. py:attribute:: verbose :value: False .. py:method:: make_optimizer() .. py:method:: make_lr_scheduler(optimizer) .. py:method:: train() .. py:method:: test(bow) .. py:method:: get_beta() .. py:method:: get_top_words(num_top_words=None) .. py:method:: export_theta()