mwptoolkit.module

• mwptoolkit.module.Attention
  • mwptoolkit.module.Attention.group_attention
    • GroupAttention
      • GroupAttention.forward()
      • GroupAttention.get_mask()
      • GroupAttention.src_to_mask()
      • GroupAttention.training
    • attention()
    • group_mask()
    • src_to_mask()
  • mwptoolkit.module.Attention.multi_head_attention
    • EPTMultiHeadAttention
      • EPTMultiHeadAttention.forward()
      • EPTMultiHeadAttention.training
    • EPTMultiHeadAttentionWeights
      • EPTMultiHeadAttentionWeights.forward()
      • EPTMultiHeadAttentionWeights.hidden_dim
      • EPTMultiHeadAttentionWeights.num_heads
      • EPTMultiHeadAttentionWeights.training
    • MultiHeadAttention
      • MultiHeadAttention.forward()
      • MultiHeadAttention.training
  • mwptoolkit.module.Attention.self_attention
    • SelfAttention
      • SelfAttention.forward()
      • SelfAttention.training
    • SelfAttentionMask
      • SelfAttentionMask.forward()
      • SelfAttentionMask.get_mask()
      • SelfAttentionMask.training
  • mwptoolkit.module.Attention.seq_attention
    • Attention
      • Attention.forward()
      • Attention.training
    • MaskedRelevantScore
      • MaskedRelevantScore.forward()
      • MaskedRelevantScore.training
    • RelevantScore
      • RelevantScore.forward()
      • RelevantScore.training
    • SeqAttention
      • SeqAttention.forward()
      • SeqAttention.training
  • mwptoolkit.module.Attention.tree_attentio
    • TreeAttention
      • TreeAttention.forward()
      • TreeAttention.training
• mwptoolkit.module.Decoder
  • mwptoolkit.module.Decoder.ept_decoder
    • AveragePooling
      • AveragePooling.extra_repr()
      • AveragePooling.forward()
      • AveragePooling.training
    • DecoderModel
      • DecoderModel._build_target_dict()
      • DecoderModel._forward_single()
      • DecoderModel._init_weights()
      • DecoderModel.forward()
      • DecoderModel.init_factor()
      • DecoderModel.is_expression_type
      • DecoderModel.required_field
      • DecoderModel.training
    • ExpressionDecoderModel
      • ExpressionDecoderModel._build_decoder_context()
      • ExpressionDecoderModel._build_decoder_input()
      • ExpressionDecoderModel._build_operand_embed()
      • ExpressionDecoderModel._forward_single()
      • ExpressionDecoderModel.embed_to_hidden
      • ExpressionDecoderModel.function_arities
      • ExpressionDecoderModel.operand_norm
      • ExpressionDecoderModel.operand_source_embedding
      • ExpressionDecoderModel.operand_source_factor
      • ExpressionDecoderModel.shared_decoder_layer
      • ExpressionDecoderModel.training
    • ExpressionPointerTransformer
      • ExpressionPointerTransformer._build_attention_keys()
      • ExpressionPointerTransformer._build_operand_embed()
      • ExpressionPointerTransformer._build_target_dict()
      • ExpressionPointerTransformer._forward_single()
      • ExpressionPointerTransformer.constant_word_embedding
      • ExpressionPointerTransformer.operand_out
      • ExpressionPointerTransformer.required_field
      • ExpressionPointerTransformer.training
    • ExpressionTransformer
      • ExpressionTransformer._build_operand_embed()
      • ExpressionTransformer._build_target_dict()
      • ExpressionTransformer._forward_single()
      • ExpressionTransformer.operand_out
      • ExpressionTransformer.operand_word_embedding
      • ExpressionTransformer.required_field
      • ExpressionTransformer.training
    • LogSoftmax
      • LogSoftmax.dim
      • LogSoftmax.forward()
    • OpDecoderModel
      • OpDecoderModel._build_decoder_context()
      • OpDecoderModel._build_decoder_input()
      • OpDecoderModel._build_word_embed()
      • OpDecoderModel._forward_single()
      • OpDecoderModel.pos_factor
      • OpDecoderModel.training
    • Squeeze
      • Squeeze.extra_repr()
      • Squeeze.forward()
      • Squeeze.training
    • VanillaOpTransformer
      • VanillaOpTransformer._build_target_dict()
      • VanillaOpTransformer._build_word_embed()
      • VanillaOpTransformer._forward_single()
      • VanillaOpTransformer.required_field
      • VanillaOpTransformer.softmax
      • VanillaOpTransformer.training
    • apply_across_dim()
    • apply_module_dict()
    • get_embedding_without_pad()
    • mask_forward()
  • mwptoolkit.module.Decoder.rnn_decoder
    • AttentionalRNNDecoder
      • AttentionalRNNDecoder.forward()
      • AttentionalRNNDecoder.init_hidden()
      • AttentionalRNNDecoder.training
    • BasicRNNDecoder
      • BasicRNNDecoder.forward()
      • BasicRNNDecoder.init_hidden()
      • BasicRNNDecoder.training
    • SalignedDecoder
      • SalignedDecoder.forward()
      • SalignedDecoder.pad_and_cat()
      • SalignedDecoder.training
  • mwptoolkit.module.Decoder.transformer_decoder
    • TransformerDecoder
      • TransformerDecoder.forward()
      • TransformerDecoder.training
  • mwptoolkit.module.Decoder.tree_decoder
    • HMSDecoder
      • HMSDecoder.forward()
      • HMSDecoder.forward_beam()
      • HMSDecoder.forward_step()
      • HMSDecoder.forward_teacher()
      • HMSDecoder.get_class_embedding_mask()
      • HMSDecoder.get_generator_embedding_mask()
      • HMSDecoder.get_mask()
      • HMSDecoder.get_pad_masks()
      • HMSDecoder.get_pointer_embedding()
      • HMSDecoder.get_pointer_mask()
      • HMSDecoder.get_pointer_meta()
      • HMSDecoder.get_predict_meta()
      • HMSDecoder.init_stacks()
      • HMSDecoder.training
    • LSTMBasedTreeDecoder
      • LSTMBasedTreeDecoder.forward()
      • LSTMBasedTreeDecoder.training
    • PredictModel
      • PredictModel.forward()
      • PredictModel.score_pn()
      • PredictModel.training
    • RNNBasedTreeDecoder
      • RNNBasedTreeDecoder.forward()
      • RNNBasedTreeDecoder.training
    • SARTreeDecoder
      • SARTreeDecoder.Semantically_Aligned_Regularization()
      • SARTreeDecoder.forward()
      • SARTreeDecoder.training
    • TreeDecoder
      • TreeDecoder.forward()
      • TreeDecoder.training
• mwptoolkit.module.Embedder
  • mwptoolkit.module.Embedder.basic_embedder
    • BasicEmbedder
      • BasicEmbedder.forward()
      • BasicEmbedder.init_embedding_params()
      • BasicEmbedder.training
  • mwptoolkit.module.Embedder.bert_embedder
    • BertEmbedder
      • BertEmbedder.forward()
      • BertEmbedder.token_resize()
      • BertEmbedder.training
  • mwptoolkit.module.Embedder.position_embedder
    • DisPositionalEncoding
      • DisPositionalEncoding.forward()
      • DisPositionalEncoding.training
    • EPTPositionalEncoding
      • EPTPositionalEncoding._forward()
      • EPTPositionalEncoding.before_trigonometric()
      • EPTPositionalEncoding.device
      • EPTPositionalEncoding.embedding_dim
      • EPTPositionalEncoding.forward()
      • EPTPositionalEncoding.training
    • PositionEmbedder
      • PositionEmbedder.forward()
      • PositionEmbedder.get_embedding()
      • PositionEmbedder.training
    • PositionEmbedder_x
      • PositionEmbedder_x.forward()
      • PositionEmbedder_x.training
    • PositionalEncoding
      • PositionalEncoding.forward()
      • PositionalEncoding.training
  • mwptoolkit.module.Embedder.roberta_embedder
    • RobertaEmbedder
      • RobertaEmbedder.forward()
      • RobertaEmbedder.token_resize()
      • RobertaEmbedder.training
• mwptoolkit.module.Encoder
  • mwptoolkit.module.Encoder.graph_based_encoder
    • GraphBasedEncoder
      • GraphBasedEncoder.forward()
      • GraphBasedEncoder.training
    • GraphBasedMultiEncoder
      • GraphBasedMultiEncoder.forward()
      • GraphBasedMultiEncoder.training
    • GraphEncoder
      • GraphEncoder.forward()
      • GraphEncoder.training
    • NumEncoder
      • NumEncoder.forward()
      • NumEncoder.training
    • replace_masked_values()
  • mwptoolkit.module.Encoder.rnn_encoder
    • BasicRNNEncoder
      • BasicRNNEncoder.forward()
      • BasicRNNEncoder.init_hidden()
      • BasicRNNEncoder.training
    • GroupAttentionRNNEncoder
      • GroupAttentionRNNEncoder.forward()
      • GroupAttentionRNNEncoder.training
    • HWCPEncoder
      • HWCPEncoder.bi_combine()
      • HWCPEncoder.clause_level_forward()
      • HWCPEncoder.dependency_encode()
      • HWCPEncoder.forward()
      • HWCPEncoder.get_mask()
      • HWCPEncoder.problem_level_forword()
      • HWCPEncoder.training
      • HWCPEncoder.word_level_forward()
    • SalignedEncoder
      • SalignedEncoder.forward()
      • SalignedEncoder.get_fix_constant()
      • SalignedEncoder.initialize_fix_constant()
      • SalignedEncoder.training
    • SelfAttentionRNNEncoder
      • SelfAttentionRNNEncoder.forward()
      • SelfAttentionRNNEncoder.init_hidden()
      • SelfAttentionRNNEncoder.training
  • mwptoolkit.module.Encoder.transformer_encoder
    • BertEncoder
      • BertEncoder.forward()
      • BertEncoder.token_resize()
      • BertEncoder.training
    • GroupATTEncoder
      • GroupATTEncoder.forward()
      • GroupATTEncoder.training
    • TransformerEncoder
      • TransformerEncoder.forward()
      • TransformerEncoder.training
• mwptoolkit.module.Environment
  • mwptoolkit.module.Environment.stack_machine
    • OPERATIONS
    • StackMachine
      • StackMachine.add_variable()
      • StackMachine.apply_embed_only()
      • StackMachine.apply_eql()
      • StackMachine.get_height()
      • StackMachine.get_solution()
      • StackMachine.get_stack()
      • StackMachine.get_top2()
      • StackMachine.push()
• mwptoolkit.module.Graph
  • mwptoolkit.module.Graph.gcn
    • GCN
      • GCN.forward()
      • GCN.training
  • mwptoolkit.module.Graph.graph_module
    • Graph_Module
      • Graph_Module.b_normal()
      • Graph_Module.forward()
      • Graph_Module.get_adj()
      • Graph_Module.normalize()
      • Graph_Module.training
    • Num_Graph_Module
      • Num_Graph_Module.forward()
      • Num_Graph_Module.normalize()
      • Num_Graph_Module.training
    • Parse_Graph_Module
      • Parse_Graph_Module.forward()
      • Parse_Graph_Module.normalize()
      • Parse_Graph_Module.training
• mwptoolkit.module.Layer
  • mwptoolkit.module.Layer.graph_layers
    • GraphConvolution
      • GraphConvolution.forward()
      • GraphConvolution.reset_parameters()
      • GraphConvolution.training
    • LayerNorm
      • LayerNorm.forward()
      • LayerNorm.training
    • MeanAggregator
      • MeanAggregator.forward()
      • MeanAggregator.training
    • PositionwiseFeedForward
      • PositionwiseFeedForward.forward()
      • PositionwiseFeedForward.training
  • mwptoolkit.module.Layer.layers
    • GenVar
      • GenVar.forward()
      • GenVar.training
    • Transformer
      • Transformer.forward()
      • Transformer.training
    • TreeAttnDecoderRNN
      • TreeAttnDecoderRNN.forward()
      • TreeAttnDecoderRNN.training
  • mwptoolkit.module.Layer.transformer_layer
    • EPTTransformerLayer
      • EPTTransformerLayer.forward()
      • EPTTransformerLayer.training
    • GAEncoderLayer
      • GAEncoderLayer.forward()
      • GAEncoderLayer.training
    • LayerNorm
      • LayerNorm.forward()
      • LayerNorm.training
    • PositionwiseFeedForward
      • PositionwiseFeedForward.forward()
      • PositionwiseFeedForward.training
    • SublayerConnection
      • SublayerConnection.forward()
      • SublayerConnection.training
    • TransformerLayer
      • TransformerLayer.forward()
      • TransformerLayer.gelu()
      • TransformerLayer.reset_parameters()
      • TransformerLayer.training
  • mwptoolkit.module.Layer.tree_layers
    • DQN
      • DQN.forward()
      • DQN.play_one()
      • DQN.training
    • Dec_LSTM
      • Dec_LSTM.forward()
      • Dec_LSTM.training
    • DecomposeModel
      • DecomposeModel.forward()
      • DecomposeModel.training
    • GateNN
      • GateNN.forward()
      • GateNN.training
    • GenerateNode
      • GenerateNode.forward()
      • GenerateNode.training
    • Merge
      • Merge.forward()
      • Merge.training
    • Node
      • Node.set_left_node()
      • Node.set_right_node()
    • NodeEmbeddingLayer
      • NodeEmbeddingLayer.forward()
      • NodeEmbeddingLayer.training
    • NodeEmbeddingNode
    • NodeGenerater
      • NodeGenerater.forward()
      • NodeGenerater.training
    • Prediction
      • Prediction.forward()
      • Prediction.training
    • RecursiveNN
      • RecursiveNN.RecurCell()
      • RecursiveNN.forward()
      • RecursiveNN.leaf_emb()
      • RecursiveNN.test()
      • RecursiveNN.test_traverse()
      • RecursiveNN.training
      • RecursiveNN.traverse()
    • Score
      • Score.forward()
      • Score.training
    • ScoreModel
      • ScoreModel.forward()
      • ScoreModel.training
    • SemanticAlignmentModule
      • SemanticAlignmentModule.forward()
      • SemanticAlignmentModule.training
    • SubTreeMerger
      • SubTreeMerger.forward()
      • SubTreeMerger.training
    • TreeAttention
      • TreeAttention.forward()
      • TreeAttention.training
    • TreeEmbedding
    • TreeEmbeddingModel
      • TreeEmbeddingModel.forward()
      • TreeEmbeddingModel.merge()
      • TreeEmbeddingModel.training
    • TreeNode
• mwptoolkit.module.Strategy
  • mwptoolkit.module.Strategy.beam_search
    • Beam
    • BeamNode
      • BeamNode.copy()
    • Beam_Search_Hypothesis
      • Beam_Search_Hypothesis.generate()
      • Beam_Search_Hypothesis.step()
      • Beam_Search_Hypothesis.stop()
    • TreeBeam
  • mwptoolkit.module.Strategy.greedy
    • greedy_search()
  • mwptoolkit.module.Strategy.sampling
    • topk_sampling()