mwptoolkit.model.Seq2Tree

• mwptoolkit.model.Seq2Tree.berttd
  • BertTD
    • BertTD.calculate_loss()
    • BertTD.convert_idx2symbol()
    • BertTD.decoder_forward()
    • BertTD.encoder_forward()
    • BertTD.forward()
    • BertTD.generate_tree_input()
    • BertTD.get_all_number_encoder_outputs()
    • BertTD.model_test()
    • BertTD.predict()
    • BertTD.training
• mwptoolkit.model.Seq2Tree.gts
  • GTS
    • GTS.calculate_loss()
    • GTS.convert_idx2symbol()
    • GTS.decoder_forward()
    • GTS.encoder_forward()
    • GTS.forward()
    • GTS.generate_tree_input()
    • GTS.get_all_number_encoder_outputs()
    • GTS.model_test()
    • GTS.predict()
    • GTS.training
• mwptoolkit.model.Seq2Tree.mwpbert
  • MWPBert
    • MWPBert.calculate_loss()
    • MWPBert.convert_idx2symbol()
    • MWPBert.decoder_forward()
    • MWPBert.encoder_forward()
    • MWPBert.forward()
    • MWPBert.generate_tree_input()
    • MWPBert.get_all_number_encoder_outputs()
    • MWPBert.model_test()
    • MWPBert.predict()
    • MWPBert.training
• mwptoolkit.model.Seq2Tree.sausolver
  • SAUSolver
    • SAUSolver.calculate_loss()
    • SAUSolver.convert_idx2symbol()
    • SAUSolver.decoder_forward()
    • SAUSolver.encoder_forward()
    • SAUSolver.evaluate_tree()
    • SAUSolver.forward()
    • SAUSolver.generate_tree_input()
    • SAUSolver.get_all_number_encoder_outputs()
    • SAUSolver.model_test()
    • SAUSolver.mse_loss()
    • SAUSolver.predict()
    • SAUSolver.train_tree()
    • SAUSolver.training
• mwptoolkit.model.Seq2Tree.treelstm
  • TreeLSTM
    • TreeLSTM.calculate_loss()
    • TreeLSTM.convert_idx2symbol()
    • TreeLSTM.copy_list()
    • TreeLSTM.decoder_forward()
    • TreeLSTM.encoder_forward()
    • TreeLSTM.forward()
    • TreeLSTM.generate_tree_input()
    • TreeLSTM.get_all_number_encoder_outputs()
    • TreeLSTM.model_test()
    • TreeLSTM.predict()
    • TreeLSTM.training
• mwptoolkit.model.Seq2Tree.trnn
  • TRNN
    • TRNN.ans_module_calculate_loss()
    • TRNN.ans_module_forward()
    • TRNN.calculate_loss()
    • TRNN.convert_idx2symbol()
    • TRNN.convert_in_idx_2_temp_idx()
    • TRNN.convert_temp_idx2symbol()
    • TRNN.convert_temp_idx_2_in_idx()
    • TRNN.forward()
    • TRNN.init_seq2seq_decoder_inputs()
    • TRNN.mask2num()
    • TRNN.model_test()
    • TRNN.predict()
    • TRNN.seq2seq_calculate_loss()
    • TRNN.seq2seq_decoder_forward()
    • TRNN.seq2seq_encoder_forward()
    • TRNN.seq2seq_forward()
    • TRNN.seq2seq_generate_t()
    • TRNN.seq2seq_generate_without_t()
    • TRNN.symbol2idx()
    • TRNN.template2tree()
    • TRNN.training
    • TRNN.tree2equation()
• mwptoolkit.model.Seq2Tree.tsn
  • TSN
    • TSN.build_graph()
    • TSN.convert_idx2symbol()
    • TSN.forward()
    • TSN.generate_tree_input()
    • TSN.get_all_number_encoder_outputs()
    • TSN.get_soft_target()
    • TSN.init_encoder_mask()
    • TSN.init_soft_target()
    • TSN.model_test()
    • TSN.predict()
    • TSN.student_calculate_loss()
    • TSN.student_net_1_decoder_forward()
    • TSN.student_net_2_decoder_forward()
    • TSN.student_net_decoder_forward()
    • TSN.student_net_encoder_forward()
    • TSN.student_net_forward()
    • TSN.student_test()
    • TSN.teacher_calculate_loss()
    • TSN.teacher_net_decoder_forward()
    • TSN.teacher_net_encoder_forward()
    • TSN.teacher_net_forward()
    • TSN.teacher_test()
    • TSN.training
  • cosine_loss()
  • cosine_sim()
  • soft_cross_entropy_loss()
  • soft_target_loss()