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