Index

_

_build_attention_keys() (mwptoolkit.module.Decoder.ept_decoder.ExpressionPointerTransformer method) _build_decoder_context() (mwptoolkit.module.Decoder.ept_decoder.ExpressionDecoderModel method) (mwptoolkit.module.Decoder.ept_decoder.OpDecoderModel method) _build_decoder_input() (mwptoolkit.module.Decoder.ept_decoder.ExpressionDecoderModel method) (mwptoolkit.module.Decoder.ept_decoder.OpDecoderModel method) _build_operand_embed() (mwptoolkit.module.Decoder.ept_decoder.ExpressionDecoderModel method) (mwptoolkit.module.Decoder.ept_decoder.ExpressionPointerTransformer method) (mwptoolkit.module.Decoder.ept_decoder.ExpressionTransformer method) _build_symbol() (mwptoolkit.data.dataset.template_dataset.TemplateDataset method) _build_target_dict() (mwptoolkit.module.Decoder.ept_decoder.DecoderModel method) (mwptoolkit.module.Decoder.ept_decoder.ExpressionPointerTransformer method) (mwptoolkit.module.Decoder.ept_decoder.ExpressionTransformer method) (mwptoolkit.module.Decoder.ept_decoder.VanillaOpTransformer method) _build_template_symbol() (mwptoolkit.data.dataset.template_dataset.TemplateDataset method) _build_word_embed() (mwptoolkit.module.Decoder.ept_decoder.OpDecoderModel method) (mwptoolkit.module.Decoder.ept_decoder.VanillaOpTransformer method)

_compute_expression_by_postfix_multi() (mwptoolkit.evaluate.evaluator.InfixEvaluator method) (mwptoolkit.evaluate.evaluator.MultiWayTreeEvaluator method) _convert_config_dict() (mwptoolkit.config.configuration.Config method) _eval_batch() (mwptoolkit.trainer.supervised_trainer.EPTTrainer method) _forward() (mwptoolkit.module.Embedder.position_embedder.EPTPositionalEncoding method) _forward_single() (mwptoolkit.module.Decoder.ept_decoder.DecoderModel method) (mwptoolkit.module.Decoder.ept_decoder.ExpressionDecoderModel method) (mwptoolkit.module.Decoder.ept_decoder.ExpressionPointerTransformer method) (mwptoolkit.module.Decoder.ept_decoder.ExpressionTransformer method) (mwptoolkit.module.Decoder.ept_decoder.OpDecoderModel method) (mwptoolkit.module.Decoder.ept_decoder.VanillaOpTransformer method) _init_weights() (mwptoolkit.module.Decoder.ept_decoder.DecoderModel method) _load_cmd_line() (mwptoolkit.config.configuration.Config method) _load_dataset() (mwptoolkit.data.dataset.abstract_dataset.AbstractDataset method) _load_fold_dataset() (mwptoolkit.data.dataset.abstract_dataset.AbstractDataset method) _normalize_gradients() (mwptoolkit.trainer.supervised_trainer.EPTTrainer method) _preprocess() (mwptoolkit.data.dataset.template_dataset.TemplateDataset method)

A

AbstractDataLoader (class in mwptoolkit.data.dataloader.abstract_dataloader) AbstractDataset (class in mwptoolkit.data.dataset.abstract_dataset) AbstractEvaluator (class in mwptoolkit.evaluate.evaluator) AbstractLoss (class in mwptoolkit.loss.abstract_loss) AbstractTrainer (class in mwptoolkit.trainer.abstract_trainer) AbstractTree (class in mwptoolkit.utils.data_structure) add_child() (mwptoolkit.utils.data_structure.Tree method) add_left_node() (mwptoolkit.utils.data_structure.DependencyNode method) add_norm() (mwptoolkit.loss.binary_cross_entropy_loss.BinaryCrossEntropyLoss method) add_right_node() (mwptoolkit.utils.data_structure.DependencyNode method) add_variable() (mwptoolkit.module.Environment.stack_machine.StackMachine method) adjust_equ() (mwptoolkit.trainer.supervised_trainer.SalignedTrainer method) alg514 (mwptoolkit.utils.enum_type.DatasetName attribute) alphabet (mwptoolkit.utils.enum_type.MaskSymbol attribute) (mwptoolkit.utils.enum_type.NumMask attribute) ans_module_calculate_loss() (mwptoolkit.model.Seq2Tree.trnn.TRNN method) ans_module_forward() (mwptoolkit.model.Seq2Tree.trnn.TRNN method) ape200k (mwptoolkit.utils.enum_type.DatasetName attribute) apply_across_dim() (in module mwptoolkit.module.Decoder.ept_decoder)

apply_embed_only() (mwptoolkit.module.Environment.stack_machine.StackMachine method) apply_eql() (mwptoolkit.module.Environment.stack_machine.StackMachine method) apply_module_dict() (in module mwptoolkit.module.Decoder.ept_decoder) ARG_CON (mwptoolkit.utils.enum_type.EPT attribute) ARG_CON_ID (mwptoolkit.utils.enum_type.EPT attribute) ARG_MEM (mwptoolkit.utils.enum_type.EPT attribute) ARG_MEM_ID (mwptoolkit.utils.enum_type.EPT attribute) ARG_NUM (mwptoolkit.utils.enum_type.EPT attribute) ARG_NUM_ID (mwptoolkit.utils.enum_type.EPT attribute) ARG_TOKENS (mwptoolkit.utils.enum_type.EPT attribute) ARG_UNK (mwptoolkit.utils.enum_type.EPT attribute) ARG_UNK_ID (mwptoolkit.utils.enum_type.EPT attribute) ARITY_MAP (mwptoolkit.utils.enum_type.EPT attribute) asdiv_a (mwptoolkit.utils.enum_type.DatasetName attribute) Attention (class in mwptoolkit.module.Attention.seq_attention) attention() (in module mwptoolkit.module.Attention.group_attention) AttentionalRNNDecoder (class in mwptoolkit.module.Decoder.rnn_decoder) attn_decoder_forward() (mwptoolkit.model.Graph2Tree.multiencdec.MultiEncDec method) AveragePooling (class in mwptoolkit.module.Decoder.ept_decoder)

B

b_normal() (mwptoolkit.module.Graph.graph_module.Graph_Module method) backward() (mwptoolkit.loss.abstract_loss.AbstractLoss method) BasicEmbedder (class in mwptoolkit.module.Embedder.basic_embedder) BasicRNNDecoder (class in mwptoolkit.module.Decoder.rnn_decoder) BasicRNNEncoder (class in mwptoolkit.module.Encoder.rnn_encoder) Beam (class in mwptoolkit.module.Strategy.beam_search) Beam_Search_Hypothesis (class in mwptoolkit.module.Strategy.beam_search) BeamNode (class in mwptoolkit.module.Strategy.beam_search) before_trigonometric() (mwptoolkit.module.Embedder.position_embedder.EPTPositionalEncoding method) BertEmbedder (class in mwptoolkit.module.Embedder.bert_embedder) BertEncoder (class in mwptoolkit.module.Encoder.transformer_encoder) BERTGen (class in mwptoolkit.model.PreTrain.bertgen) BertTD (class in mwptoolkit.model.Seq2Tree.berttd) BertTDTrainer (class in mwptoolkit.trainer.supervised_trainer)

bi_combine() (mwptoolkit.module.Encoder.rnn_encoder.HWCPEncoder method) BinaryCrossEntropyLoss (class in mwptoolkit.loss.binary_cross_entropy_loss) BinaryTree (class in mwptoolkit.utils.data_structure) BRG_TOKEN (mwptoolkit.utils.enum_type.SpecialTokens attribute) build_batch_for_predict() (mwptoolkit.data.dataloader.dataloader_ept.DataLoaderEPT method) (mwptoolkit.data.dataloader.dataloader_hms.DataLoaderHMS method) (mwptoolkit.data.dataloader.dataloader_multiencdec.DataLoaderMultiEncDec method) (mwptoolkit.data.dataloader.multi_equation_dataloader.MultiEquationDataLoader method) (mwptoolkit.data.dataloader.pretrain_dataloader.PretrainDataLoader method) (mwptoolkit.data.dataloader.single_equation_dataloader.SingleEquationDataLoader method) build_graph() (mwptoolkit.model.Graph2Tree.graph2tree.Graph2Tree method) (mwptoolkit.model.Seq2Tree.tsn.TSN method) build_pos_to_file_with_pyltp() (mwptoolkit.data.dataset.dataset_multiencdec.DatasetMultiEncDec method) build_pos_to_file_with_stanza() (mwptoolkit.data.dataset.dataset_multiencdec.DatasetMultiEncDec method)

C

calculate_loss() (mwptoolkit.model.Graph2Tree.graph2tree.Graph2Tree method) (mwptoolkit.model.Graph2Tree.multiencdec.MultiEncDec method) (mwptoolkit.model.PreTrain.bertgen.BERTGen method) (mwptoolkit.model.PreTrain.gpt2.GPT2 method) (mwptoolkit.model.PreTrain.robertagen.RobertaGen method) (mwptoolkit.model.Seq2Seq.dns.DNS method) (mwptoolkit.model.Seq2Seq.ept.EPT method) (mwptoolkit.model.Seq2Seq.groupatt.GroupATT method) (mwptoolkit.model.Seq2Seq.mathen.MathEN method) (mwptoolkit.model.Seq2Seq.rnnencdec.RNNEncDec method) (mwptoolkit.model.Seq2Seq.rnnvae.RNNVAE method) (mwptoolkit.model.Seq2Seq.saligned.Saligned method) (mwptoolkit.model.Seq2Seq.transformer.Transformer method) (mwptoolkit.model.Seq2Tree.berttd.BertTD method) (mwptoolkit.model.Seq2Tree.gts.GTS method) (mwptoolkit.model.Seq2Tree.mwpbert.MWPBert method) (mwptoolkit.model.Seq2Tree.sausolver.SAUSolver method) (mwptoolkit.model.Seq2Tree.treelstm.TreeLSTM method) (mwptoolkit.model.Seq2Tree.trnn.TRNN method) clause_level_forward() (mwptoolkit.module.Encoder.rnn_encoder.HWCPEncoder method) clones() (in module mwptoolkit.utils.utils) CON_PREFIX (mwptoolkit.utils.enum_type.EPT attribute) Config (class in mwptoolkit.config.configuration) constant_number() (in module mwptoolkit.utils.preprocess_tool.number_operator) constant_word_embedding (mwptoolkit.module.Decoder.ept_decoder.ExpressionPointerTransformer attribute) convert_idx2symbol() (mwptoolkit.model.Graph2Tree.graph2tree.Graph2Tree method) (mwptoolkit.model.PreTrain.bertgen.BERTGen method) (mwptoolkit.model.PreTrain.gpt2.GPT2 method) (mwptoolkit.model.PreTrain.robertagen.RobertaGen method) (mwptoolkit.model.Seq2Seq.dns.DNS method) (mwptoolkit.model.Seq2Seq.ept.EPT method) (mwptoolkit.model.Seq2Seq.groupatt.GroupATT method) (mwptoolkit.model.Seq2Seq.mathen.MathEN method) (mwptoolkit.model.Seq2Seq.rnnencdec.RNNEncDec method) (mwptoolkit.model.Seq2Seq.rnnvae.RNNVAE method) (mwptoolkit.model.Seq2Seq.saligned.Saligned method) (mwptoolkit.model.Seq2Seq.transformer.Transformer method) (mwptoolkit.model.Seq2Tree.berttd.BertTD method) (mwptoolkit.model.Seq2Tree.gts.GTS method) (mwptoolkit.model.Seq2Tree.mwpbert.MWPBert method) (mwptoolkit.model.Seq2Tree.sausolver.SAUSolver method) (mwptoolkit.model.Seq2Tree.treelstm.TreeLSTM method) (mwptoolkit.model.Seq2Tree.trnn.TRNN method) (mwptoolkit.model.Seq2Tree.tsn.TSN method)

convert_idx2symbol1() (mwptoolkit.model.Graph2Tree.multiencdec.MultiEncDec method) convert_idx2symbol2() (mwptoolkit.model.Graph2Tree.multiencdec.MultiEncDec method) convert_idx_2_symbol() (mwptoolkit.data.dataloader.abstract_dataloader.AbstractDataLoader method) convert_idx_2_word() (mwptoolkit.data.dataloader.abstract_dataloader.AbstractDataLoader method) convert_in_idx_2_out_idx() (mwptoolkit.model.Seq2Seq.dns.DNS method) (mwptoolkit.model.Seq2Seq.groupatt.GroupATT method) (mwptoolkit.model.Seq2Seq.mathen.MathEN method) (mwptoolkit.model.Seq2Seq.rnnencdec.RNNEncDec method) (mwptoolkit.model.Seq2Seq.rnnvae.RNNVAE method) (mwptoolkit.model.Seq2Seq.transformer.Transformer method) convert_in_idx_2_temp_idx() (mwptoolkit.model.Seq2Tree.trnn.TRNN method) convert_mask_num() (mwptoolkit.model.Seq2Seq.saligned.Saligned method) convert_out_idx_2_in_idx() (mwptoolkit.model.Seq2Seq.dns.DNS method) (mwptoolkit.model.Seq2Seq.groupatt.GroupATT method) (mwptoolkit.model.Seq2Seq.mathen.MathEN method) (mwptoolkit.model.Seq2Seq.rnnencdec.RNNEncDec method) (mwptoolkit.model.Seq2Seq.rnnvae.RNNVAE method) (mwptoolkit.model.Seq2Seq.transformer.Transformer method) convert_symbol_2_idx() (mwptoolkit.data.dataloader.abstract_dataloader.AbstractDataLoader method) convert_temp_idx2symbol() (mwptoolkit.model.Seq2Tree.trnn.TRNN method) convert_temp_idx_2_in_idx() (mwptoolkit.model.Seq2Tree.trnn.TRNN method) convert_word_2_idx() (mwptoolkit.data.dataloader.abstract_dataloader.AbstractDataLoader method) copy() (mwptoolkit.module.Strategy.beam_search.BeamNode method) copy_list() (in module mwptoolkit.utils.utils) (mwptoolkit.model.Seq2Tree.treelstm.TreeLSTM method) cosine_loss() (in module mwptoolkit.model.Seq2Tree.tsn) cosine_sim() (in module mwptoolkit.model.Seq2Tree.tsn) create_dataloader() (in module mwptoolkit.data.utils) create_dataset() (in module mwptoolkit.data.utils) cross_validation_load() (mwptoolkit.data.dataset.abstract_dataset.AbstractDataset method) CrossEntropyLoss (class in mwptoolkit.loss.cross_entropy_loss)

D

DataLoaderEPT (class in mwptoolkit.data.dataloader.dataloader_ept) DataLoaderHMS (class in mwptoolkit.data.dataloader.dataloader_hms) DataLoaderMultiEncDec (class in mwptoolkit.data.dataloader.dataloader_multiencdec) dataset_load() (mwptoolkit.data.dataset.abstract_dataset.AbstractDataset method) DatasetEPT (class in mwptoolkit.data.dataset.dataset_ept) DatasetHMS (class in mwptoolkit.data.dataset.dataset_hms) DatasetLanguage (class in mwptoolkit.utils.enum_type) DatasetMultiEncDec (class in mwptoolkit.data.dataset.dataset_multiencdec) DatasetName (class in mwptoolkit.utils.enum_type) DatasetType (class in mwptoolkit.utils.enum_type) Dec_LSTM (class in mwptoolkit.module.Layer.tree_layers) decode() (mwptoolkit.model.PreTrain.bertgen.BERTGen method) (mwptoolkit.model.PreTrain.robertagen.RobertaGen method) (mwptoolkit.model.Seq2Seq.dns.DNS method) (mwptoolkit.model.Seq2Seq.ept.EPT method) (mwptoolkit.model.Seq2Seq.groupatt.GroupATT method) (mwptoolkit.model.Seq2Seq.mathen.MathEN method) (mwptoolkit.model.Seq2Seq.rnnencdec.RNNEncDec method) (mwptoolkit.model.Seq2Seq.rnnvae.RNNVAE method) (mwptoolkit.model.Seq2Seq.transformer.Transformer method) decode_() (mwptoolkit.model.PreTrain.bertgen.BERTGen method) (mwptoolkit.model.PreTrain.gpt2.GPT2 method) (mwptoolkit.model.PreTrain.robertagen.RobertaGen method) decoder_forward() (mwptoolkit.model.Graph2Tree.graph2tree.Graph2Tree method) (mwptoolkit.model.Graph2Tree.multiencdec.MultiEncDec method) (mwptoolkit.model.PreTrain.bertgen.BERTGen method) (mwptoolkit.model.PreTrain.gpt2.GPT2 method) (mwptoolkit.model.PreTrain.robertagen.RobertaGen method) (mwptoolkit.model.Seq2Seq.dns.DNS method) (mwptoolkit.model.Seq2Seq.ept.EPT method) (mwptoolkit.model.Seq2Seq.groupatt.GroupATT method) (mwptoolkit.model.Seq2Seq.mathen.MathEN method) (mwptoolkit.model.Seq2Seq.rnnencdec.RNNEncDec method) (mwptoolkit.model.Seq2Seq.rnnvae.RNNVAE method) (mwptoolkit.model.Seq2Seq.saligned.Saligned method) (mwptoolkit.model.Seq2Seq.transformer.Transformer method) (mwptoolkit.model.Seq2Tree.berttd.BertTD method) (mwptoolkit.model.Seq2Tree.gts.GTS method) (mwptoolkit.model.Seq2Tree.mwpbert.MWPBert method) (mwptoolkit.model.Seq2Tree.sausolver.SAUSolver method) (mwptoolkit.model.Seq2Tree.treelstm.TreeLSTM method)

DecoderModel (class in mwptoolkit.module.Decoder.ept_decoder) DecomposeModel (class in mwptoolkit.module.Layer.tree_layers) dependency_encode() (mwptoolkit.module.Encoder.rnn_encoder.HWCPEncoder method) DependencyNode (class in mwptoolkit.utils.data_structure) DependencyTree (class in mwptoolkit.utils.data_structure) deprel_tree_to_file() (in module mwptoolkit.utils.preprocess_tool.sentence_operator) device (mwptoolkit.module.Embedder.position_embedder.EPTPositionalEncoding property) dim (mwptoolkit.module.Decoder.ept_decoder.LogSoftmax attribute) DisPositionalEncoding (class in mwptoolkit.module.Embedder.position_embedder) DNS (class in mwptoolkit.model.Seq2Seq.dns) DQN (class in mwptoolkit.module.Layer.tree_layers) draw (mwptoolkit.utils.enum_type.DatasetName attribute)

E

embed_to_hidden (mwptoolkit.module.Decoder.ept_decoder.ExpressionDecoderModel attribute) embedding_dim (mwptoolkit.module.Embedder.position_embedder.EPTPositionalEncoding attribute) en (mwptoolkit.utils.enum_type.DatasetLanguage attribute) en_rule1_process() (mwptoolkit.data.dataset.abstract_dataset.AbstractDataset method) EN_rule1_stat() (in module mwptoolkit.utils.preprocess_tool.equation_operator) EN_rule2() (in module mwptoolkit.utils.preprocess_tool.equation_operator) en_rule2_process() (mwptoolkit.data.dataset.abstract_dataset.AbstractDataset method) encode_() (mwptoolkit.model.PreTrain.gpt2.GPT2 method) encoder_forward() (mwptoolkit.model.Graph2Tree.graph2tree.Graph2Tree method) (mwptoolkit.model.Graph2Tree.multiencdec.MultiEncDec method) (mwptoolkit.model.PreTrain.bertgen.BERTGen method) (mwptoolkit.model.PreTrain.robertagen.RobertaGen method) (mwptoolkit.model.Seq2Seq.dns.DNS method) (mwptoolkit.model.Seq2Seq.ept.EPT method) (mwptoolkit.model.Seq2Seq.groupatt.GroupATT method) (mwptoolkit.model.Seq2Seq.mathen.MathEN method) (mwptoolkit.model.Seq2Seq.rnnencdec.RNNEncDec method) (mwptoolkit.model.Seq2Seq.rnnvae.RNNVAE method) (mwptoolkit.model.Seq2Seq.saligned.Saligned method) (mwptoolkit.model.Seq2Seq.transformer.Transformer method) (mwptoolkit.model.Seq2Tree.berttd.BertTD method) (mwptoolkit.model.Seq2Tree.gts.GTS method) (mwptoolkit.model.Seq2Tree.mwpbert.MWPBert method) (mwptoolkit.model.Seq2Tree.sausolver.SAUSolver method) (mwptoolkit.model.Seq2Tree.treelstm.TreeLSTM method) english_word_2_num() (in module mwptoolkit.utils.preprocess_tool.number_operator) EOS_TOKEN (mwptoolkit.utils.enum_type.SpecialTokens attribute) EPT (class in mwptoolkit.model.Seq2Seq.ept) (class in mwptoolkit.utils.enum_type) ept_preprocess() (in module mwptoolkit.utils.preprocess_tool.dataset_operator) EPTMultiHeadAttention (class in mwptoolkit.module.Attention.multi_head_attention) EPTMultiHeadAttentionWeights (class in mwptoolkit.module.Attention.multi_head_attention)

EPTPositionalEncoding (class in mwptoolkit.module.Embedder.position_embedder) EPTTrainer (class in mwptoolkit.trainer.supervised_trainer) EPTTransformerLayer (class in mwptoolkit.module.Layer.transformer_layer) equ2tree() (mwptoolkit.utils.data_structure.AbstractTree method) (mwptoolkit.utils.data_structure.BinaryTree method) (mwptoolkit.utils.data_structure.GoldTree method) equ2tree_() (mwptoolkit.utils.data_structure.BinaryTree method) eval_batch() (mwptoolkit.loss.abstract_loss.AbstractLoss method) (mwptoolkit.loss.binary_cross_entropy_loss.BinaryCrossEntropyLoss method) (mwptoolkit.loss.cross_entropy_loss.CrossEntropyLoss method) (mwptoolkit.loss.masked_cross_entropy_loss.MaskedCrossEntropyLoss method) (mwptoolkit.loss.mse_loss.MSELoss method) (mwptoolkit.loss.nll_loss.NLLLoss method) (mwptoolkit.loss.smoothed_cross_entropy_loss.SmoothCrossEntropyLoss method) eval_source() (mwptoolkit.evaluate.evaluator.PostfixEvaluator method) (mwptoolkit.evaluate.evaluator.PrefixEvaluator method) evaluate() (mwptoolkit.trainer.abstract_trainer.AbstractTrainer method) (mwptoolkit.trainer.supervised_trainer.EPTTrainer method) (mwptoolkit.trainer.supervised_trainer.GTSTrainer method) (mwptoolkit.trainer.supervised_trainer.SalignedTrainer method) (mwptoolkit.trainer.supervised_trainer.SupervisedTrainer method) (mwptoolkit.trainer.supervised_trainer.TreeLSTMTrainer method) (mwptoolkit.trainer.supervised_trainer.TRNNTrainer method) (mwptoolkit.trainer.template_trainer.TemplateTrainer method) evaluate_student() (mwptoolkit.trainer.supervised_trainer.TSNTrainer method) evaluate_teacher() (mwptoolkit.trainer.supervised_trainer.TSNTrainer method) evaluate_tree() (mwptoolkit.model.Seq2Tree.sausolver.SAUSolver method) ExpressionDecoderModel (class in mwptoolkit.module.Decoder.ept_decoder) ExpressionPointerTransformer (class in mwptoolkit.module.Decoder.ept_decoder) ExpressionTransformer (class in mwptoolkit.module.Decoder.ept_decoder) extra_repr() (mwptoolkit.module.Decoder.ept_decoder.AveragePooling method) (mwptoolkit.module.Decoder.ept_decoder.Squeeze method)

F

FIELD_EXPR_GEN (mwptoolkit.utils.enum_type.EPT attribute) FIELD_EXPR_PTR (mwptoolkit.utils.enum_type.EPT attribute) FIELD_OP_GEN (mwptoolkit.utils.enum_type.EPT attribute) filter_END() (mwptoolkit.model.Seq2Seq.dns.DNS method) filter_op() (mwptoolkit.model.Seq2Seq.dns.DNS method) find_ept_numbers_in_text() (in module mwptoolkit.utils.preprocess_tool.sentence_operator) fit() (mwptoolkit.trainer.abstract_trainer.AbstractTrainer method) (mwptoolkit.trainer.supervised_trainer.EPTTrainer method) (mwptoolkit.trainer.supervised_trainer.GTSTrainer method) (mwptoolkit.trainer.supervised_trainer.SalignedTrainer method) (mwptoolkit.trainer.supervised_trainer.SupervisedTrainer method) (mwptoolkit.trainer.supervised_trainer.TreeLSTMTrainer method) (mwptoolkit.trainer.supervised_trainer.TRNNTrainer method) (mwptoolkit.trainer.supervised_trainer.TSNTrainer method) (mwptoolkit.trainer.template_trainer.TemplateTrainer method) fix_process() (mwptoolkit.data.dataset.abstract_dataset.AbstractDataset method) FixType (class in mwptoolkit.utils.enum_type) FOLLOWING_ZERO_PATTERN (mwptoolkit.utils.enum_type.EPT attribute) FORMAT_MEM (mwptoolkit.utils.enum_type.EPT attribute) FORMAT_NUM (mwptoolkit.utils.enum_type.EPT attribute) FORMAT_VAR (mwptoolkit.utils.enum_type.EPT attribute) forward() (mwptoolkit.loss.smoothed_cross_entropy_loss.SmoothedCrossEntropyLoss method) (mwptoolkit.model.Graph2Tree.graph2tree.Graph2Tree method) (mwptoolkit.model.Graph2Tree.multiencdec.MultiEncDec method) (mwptoolkit.model.PreTrain.bertgen.BERTGen method) (mwptoolkit.model.PreTrain.gpt2.GPT2 method) (mwptoolkit.model.PreTrain.robertagen.RobertaGen method) (mwptoolkit.model.Seq2Seq.dns.DNS method) (mwptoolkit.model.Seq2Seq.ept.EPT method) (mwptoolkit.model.Seq2Seq.groupatt.GroupATT method) (mwptoolkit.model.Seq2Seq.mathen.MathEN method) (mwptoolkit.model.Seq2Seq.rnnencdec.RNNEncDec method) (mwptoolkit.model.Seq2Seq.rnnvae.RNNVAE method) (mwptoolkit.model.Seq2Seq.saligned.Saligned method) (mwptoolkit.model.Seq2Seq.transformer.Transformer method) (mwptoolkit.model.Seq2Tree.berttd.BertTD method) (mwptoolkit.model.Seq2Tree.gts.GTS method) (mwptoolkit.model.Seq2Tree.mwpbert.MWPBert method) (mwptoolkit.model.Seq2Tree.sausolver.SAUSolver method) (mwptoolkit.model.Seq2Tree.treelstm.TreeLSTM method) (mwptoolkit.model.Seq2Tree.trnn.TRNN method) (mwptoolkit.model.Seq2Tree.tsn.TSN method) (mwptoolkit.module.Attention.group_attention.GroupAttention method) (mwptoolkit.module.Attention.multi_head_attention.EPTMultiHeadAttention method) (mwptoolkit.module.Attention.multi_head_attention.EPTMultiHeadAttentionWeights method) (mwptoolkit.module.Attention.multi_head_attention.MultiHeadAttention method) (mwptoolkit.module.Attention.self_attention.SelfAttention method) (mwptoolkit.module.Attention.self_attention.SelfAttentionMask method) (mwptoolkit.module.Attention.seq_attention.Attention method) (mwptoolkit.module.Attention.seq_attention.MaskedRelevantScore method) (mwptoolkit.module.Attention.seq_attention.RelevantScore method) (mwptoolkit.module.Attention.seq_attention.SeqAttention method) (mwptoolkit.module.Attention.tree_attention.TreeAttention method) (mwptoolkit.module.Decoder.ept_decoder.AveragePooling method) (mwptoolkit.module.Decoder.ept_decoder.DecoderModel method) (mwptoolkit.module.Decoder.ept_decoder.LogSoftmax method) (mwptoolkit.module.Decoder.ept_decoder.Squeeze method) (mwptoolkit.module.Decoder.rnn_decoder.AttentionalRNNDecoder method) (mwptoolkit.module.Decoder.rnn_decoder.BasicRNNDecoder method) (mwptoolkit.module.Decoder.rnn_decoder.SalignedDecoder method) (mwptoolkit.module.Decoder.transformer_decoder.TransformerDecoder method) (mwptoolkit.module.Decoder.tree_decoder.HMSDecoder method) (mwptoolkit.module.Decoder.tree_decoder.LSTMBasedTreeDecoder method) (mwptoolkit.module.Decoder.tree_decoder.PredictModel method) (mwptoolkit.module.Decoder.tree_decoder.RNNBasedTreeDecoder method) (mwptoolkit.module.Decoder.tree_decoder.SARTreeDecoder method) (mwptoolkit.module.Decoder.tree_decoder.TreeDecoder method) (mwptoolkit.module.Embedder.basic_embedder.BasicEmbedder method) (mwptoolkit.module.Embedder.bert_embedder.BertEmbedder method) (mwptoolkit.module.Embedder.position_embedder.DisPositionalEncoding method) (mwptoolkit.module.Embedder.position_embedder.EPTPositionalEncoding method) (mwptoolkit.module.Embedder.position_embedder.PositionalEncoding method) (mwptoolkit.module.Embedder.position_embedder.PositionEmbedder method) (mwptoolkit.module.Embedder.position_embedder.PositionEmbedder_x method) (mwptoolkit.module.Embedder.roberta_embedder.RobertaEmbedder method) (mwptoolkit.module.Encoder.graph_based_encoder.GraphBasedEncoder method) (mwptoolkit.module.Encoder.graph_based_encoder.GraphBasedMultiEncoder method) (mwptoolkit.module.Encoder.graph_based_encoder.GraphEncoder method) (mwptoolkit.module.Encoder.graph_based_encoder.NumEncoder method) (mwptoolkit.module.Encoder.rnn_encoder.BasicRNNEncoder method) (mwptoolkit.module.Encoder.rnn_encoder.GroupAttentionRNNEncoder method) (mwptoolkit.module.Encoder.rnn_encoder.HWCPEncoder method) (mwptoolkit.module.Encoder.rnn_encoder.SalignedEncoder method) (mwptoolkit.module.Encoder.rnn_encoder.SelfAttentionRNNEncoder method) (mwptoolkit.module.Encoder.transformer_encoder.BertEncoder method) (mwptoolkit.module.Encoder.transformer_encoder.GroupATTEncoder method) (mwptoolkit.module.Encoder.transformer_encoder.TransformerEncoder method) (mwptoolkit.module.Graph.gcn.GCN method) (mwptoolkit.module.Graph.graph_module.Graph_Module method) (mwptoolkit.module.Graph.graph_module.Num_Graph_Module method) (mwptoolkit.module.Graph.graph_module.Parse_Graph_Module method) (mwptoolkit.module.Layer.graph_layers.GraphConvolution method) (mwptoolkit.module.Layer.graph_layers.LayerNorm method) (mwptoolkit.module.Layer.graph_layers.MeanAggregator method) (mwptoolkit.module.Layer.graph_layers.PositionwiseFeedForward method) (mwptoolkit.module.Layer.layers.GenVar method) (mwptoolkit.module.Layer.layers.Transformer method) (mwptoolkit.module.Layer.layers.TreeAttnDecoderRNN method) (mwptoolkit.module.Layer.transformer_layer.EPTTransformerLayer method) (mwptoolkit.module.Layer.transformer_layer.GAEncoderLayer method) (mwptoolkit.module.Layer.transformer_layer.LayerNorm method) (mwptoolkit.module.Layer.transformer_layer.PositionwiseFeedForward method) (mwptoolkit.module.Layer.transformer_layer.SublayerConnection method) (mwptoolkit.module.Layer.transformer_layer.TransformerLayer method) (mwptoolkit.module.Layer.tree_layers.Dec_LSTM method) (mwptoolkit.module.Layer.tree_layers.DecomposeModel method) (mwptoolkit.module.Layer.tree_layers.DQN method) (mwptoolkit.module.Layer.tree_layers.GateNN method) (mwptoolkit.module.Layer.tree_layers.GenerateNode method) (mwptoolkit.module.Layer.tree_layers.Merge method) (mwptoolkit.module.Layer.tree_layers.NodeEmbeddingLayer method) (mwptoolkit.module.Layer.tree_layers.NodeGenerater method) (mwptoolkit.module.Layer.tree_layers.Prediction method) (mwptoolkit.module.Layer.tree_layers.RecursiveNN method) (mwptoolkit.module.Layer.tree_layers.Score method) (mwptoolkit.module.Layer.tree_layers.ScoreModel method) (mwptoolkit.module.Layer.tree_layers.SemanticAlignmentModule method) (mwptoolkit.module.Layer.tree_layers.SubTreeMerger method) (mwptoolkit.module.Layer.tree_layers.TreeAttention method) (mwptoolkit.module.Layer.tree_layers.TreeEmbeddingModel method)

forward_beam() (mwptoolkit.module.Decoder.tree_decoder.HMSDecoder method) forward_step() (mwptoolkit.module.Decoder.tree_decoder.HMSDecoder method) forward_teacher() (mwptoolkit.module.Decoder.tree_decoder.HMSDecoder method) fraction_word_to_num() (in module mwptoolkit.utils.preprocess_tool.number_operator) FRACTIONAL_PATTERN (mwptoolkit.utils.enum_type.EPT attribute) from_infix_to_multi_way_tree() (in module mwptoolkit.utils.preprocess_tool.equation_operator) from_infix_to_postfix() (in module mwptoolkit.utils.preprocess_tool.equation_operator) from_infix_to_prefix() (in module mwptoolkit.utils.preprocess_tool.equation_operator) from_postfix_to_infix() (in module mwptoolkit.utils.preprocess_tool.equation_operator) from_postfix_to_prefix() (in module mwptoolkit.utils.preprocess_tool.equation_operator) from_prefix_to_infix() (in module mwptoolkit.utils.preprocess_tool.equation_operator) from_prefix_to_postfix() (in module mwptoolkit.utils.preprocess_tool.equation_operator) fully_supervised (mwptoolkit.utils.enum_type.SupervisingMode attribute) FUN_END_EQN (mwptoolkit.utils.enum_type.EPT attribute) FUN_END_EQN_ID (mwptoolkit.utils.enum_type.EPT attribute) FUN_EQ_SGN_ID (mwptoolkit.utils.enum_type.EPT attribute) FUN_NEW_EQN (mwptoolkit.utils.enum_type.EPT attribute) FUN_NEW_EQN_ID (mwptoolkit.utils.enum_type.EPT attribute) FUN_NEW_VAR (mwptoolkit.utils.enum_type.EPT attribute) FUN_NEW_VAR_ID (mwptoolkit.utils.enum_type.EPT attribute) FUN_TOKENS (mwptoolkit.utils.enum_type.EPT attribute) FUN_TOKENS_WITH_EQ (mwptoolkit.utils.enum_type.EPT attribute) function_arities (mwptoolkit.module.Decoder.ept_decoder.ExpressionDecoderModel attribute)

G

GAEncoderLayer (class in mwptoolkit.module.Layer.transformer_layer) GateNN (class in mwptoolkit.module.Layer.tree_layers) gather_vectors() (mwptoolkit.model.Seq2Seq.ept.EPT method) GCN (class in mwptoolkit.module.Graph.gcn) gelu() (mwptoolkit.module.Layer.transformer_layer.TransformerLayer method) generate() (mwptoolkit.module.Strategy.beam_search.Beam_Search_Hypothesis method) generate_decoder_input() (mwptoolkit.model.Graph2Tree.multiencdec.MultiEncDec method) generate_tree_input() (mwptoolkit.model.Graph2Tree.graph2tree.Graph2Tree method) (mwptoolkit.model.Graph2Tree.multiencdec.MultiEncDec method) (mwptoolkit.model.Seq2Tree.berttd.BertTD method) (mwptoolkit.model.Seq2Tree.gts.GTS method) (mwptoolkit.model.Seq2Tree.mwpbert.MWPBert method) (mwptoolkit.model.Seq2Tree.sausolver.SAUSolver method) (mwptoolkit.model.Seq2Tree.treelstm.TreeLSTM method) (mwptoolkit.model.Seq2Tree.tsn.TSN method) GenerateNode (class in mwptoolkit.module.Layer.tree_layers) GenVar (class in mwptoolkit.module.Layer.layers) get_adj() (mwptoolkit.module.Graph.graph_module.Graph_Module method) get_all_number_encoder_outputs() (mwptoolkit.model.Graph2Tree.graph2tree.Graph2Tree method) (mwptoolkit.model.Graph2Tree.multiencdec.MultiEncDec method) (mwptoolkit.model.Seq2Tree.berttd.BertTD method) (mwptoolkit.model.Seq2Tree.gts.GTS method) (mwptoolkit.model.Seq2Tree.mwpbert.MWPBert method) (mwptoolkit.model.Seq2Tree.sausolver.SAUSolver method) (mwptoolkit.model.Seq2Tree.treelstm.TreeLSTM method) (mwptoolkit.model.Seq2Tree.tsn.TSN method) get_class_embedding_mask() (mwptoolkit.module.Decoder.tree_decoder.HMSDecoder method) get_dataloader_module() (in module mwptoolkit.data.utils) get_dataset_module() (in module mwptoolkit.data.utils) get_deprel_tree() (in module mwptoolkit.utils.preprocess_tool.sentence_operator) get_deprel_tree_() (in module mwptoolkit.utils.preprocess_tool.sentence_operator) get_embedding() (mwptoolkit.module.Embedder.position_embedder.PositionEmbedder method) get_embedding_without_pad() (in module mwptoolkit.module.Decoder.ept_decoder) get_evaluator() (in module mwptoolkit.evaluate.evaluator) get_evaluator_module() (in module mwptoolkit.evaluate.evaluator) get_fix_constant() (mwptoolkit.module.Encoder.rnn_encoder.SalignedEncoder method) get_generator_embedding_mask() (mwptoolkit.module.Decoder.tree_decoder.HMSDecoder method) get_group_nums() (in module mwptoolkit.utils.preprocess_tool.sentence_operator) get_group_nums_() (in module mwptoolkit.utils.preprocess_tool.sentence_operator) get_height() (mwptoolkit.module.Environment.stack_machine.StackMachine method) get_loss() (mwptoolkit.loss.abstract_loss.AbstractLoss method) (mwptoolkit.loss.binary_cross_entropy_loss.BinaryCrossEntropyLoss method) (mwptoolkit.loss.cross_entropy_loss.CrossEntropyLoss method) (mwptoolkit.loss.masked_cross_entropy_loss.MaskedCrossEntropyLoss method) (mwptoolkit.loss.mse_loss.MSELoss method) (mwptoolkit.loss.nll_loss.NLLLoss method) (mwptoolkit.loss.smoothed_cross_entropy_loss.SmoothCrossEntropyLoss method) get_mask() (mwptoolkit.module.Attention.group_attention.GroupAttention method) (mwptoolkit.module.Attention.self_attention.SelfAttentionMask static method) (mwptoolkit.module.Decoder.tree_decoder.HMSDecoder method) (mwptoolkit.module.Encoder.rnn_encoder.HWCPEncoder method)

get_model() (in module mwptoolkit.utils.utils) get_num_mask() (in module mwptoolkit.data.dataloader.dataloader_hms) (in module mwptoolkit.data.dataloader.multi_equation_dataloader) (in module mwptoolkit.data.dataloader.pretrain_dataloader) (in module mwptoolkit.data.dataloader.single_equation_dataloader) get_num_pos() (in module mwptoolkit.utils.preprocess_tool.number_transfer) get_pad_masks() (mwptoolkit.module.Decoder.tree_decoder.HMSDecoder method) get_parse_graph_batch() (mwptoolkit.data.dataloader.dataloader_multiencdec.DataLoaderMultiEncDec method) get_pointer_embedding() (mwptoolkit.module.Decoder.tree_decoder.HMSDecoder method) get_pointer_mask() (mwptoolkit.module.Decoder.tree_decoder.HMSDecoder method) get_pointer_meta() (mwptoolkit.module.Decoder.tree_decoder.HMSDecoder method) get_predict_meta() (mwptoolkit.module.Decoder.tree_decoder.HMSDecoder method) get_result() (mwptoolkit.evaluate.evaluator.Solver method) get_soft_target() (mwptoolkit.model.Seq2Tree.tsn.TSN method) get_solution() (mwptoolkit.module.Environment.stack_machine.StackMachine method) get_span_level_deprel_tree() (in module mwptoolkit.utils.preprocess_tool.sentence_operator) get_span_level_deprel_tree_() (in module mwptoolkit.utils.preprocess_tool.sentence_operator) get_stack() (mwptoolkit.module.Environment.stack_machine.StackMachine method) get_top2() (mwptoolkit.module.Environment.stack_machine.StackMachine method) get_trainer() (in module mwptoolkit.utils.utils) get_trainer_() (in module mwptoolkit.utils.utils) get_vocab_size() (mwptoolkit.data.dataset.dataset_ept.DatasetEPT method) (mwptoolkit.data.dataset.dataset_hms.DatasetHMS method) (mwptoolkit.data.dataset.multi_equation_dataset.MultiEquationDataset method) (mwptoolkit.data.dataset.pretrain_dataset.PretrainDataset method) (mwptoolkit.data.dataset.single_equation_dataset.SingleEquationDataset method) (mwptoolkit.data.dataset.template_dataset.TemplateDataset method) get_weakly_supervised() (in module mwptoolkit.utils.utils) GoldTree (class in mwptoolkit.utils.data_structure) GPT2 (class in mwptoolkit.model.PreTrain.gpt2) Graph2Tree (class in mwptoolkit.model.Graph2Tree.graph2tree) Graph2TreeTrainer (class in mwptoolkit.trainer.supervised_trainer) Graph_Module (class in mwptoolkit.module.Graph.graph_module) GraphBasedEncoder (class in mwptoolkit.module.Encoder.graph_based_encoder) GraphBasedMultiEncoder (class in mwptoolkit.module.Encoder.graph_based_encoder) GraphConvolution (class in mwptoolkit.module.Layer.graph_layers) GraphEncoder (class in mwptoolkit.module.Encoder.graph_based_encoder) greedy_search() (in module mwptoolkit.module.Strategy.greedy) group_mask() (in module mwptoolkit.module.Attention.group_attention) GroupATT (class in mwptoolkit.model.Seq2Seq.groupatt) GroupATTEncoder (class in mwptoolkit.module.Encoder.transformer_encoder) GroupAttention (class in mwptoolkit.module.Attention.group_attention) GroupAttentionRNNEncoder (class in mwptoolkit.module.Encoder.rnn_encoder) GTS (class in mwptoolkit.model.Seq2Tree.gts) GTSTrainer (class in mwptoolkit.trainer.supervised_trainer)

H

hidden_dim (mwptoolkit.module.Attention.multi_head_attention.EPTMultiHeadAttentionWeights property) HMSDecoder (class in mwptoolkit.module.Decoder.tree_decoder) HMSTrainer (class in mwptoolkit.trainer.supervised_trainer)

hmwp (mwptoolkit.utils.enum_type.DatasetName attribute) HWCPEncoder (class in mwptoolkit.module.Encoder.rnn_encoder) hyper_search_process() (in module mwptoolkit.hyper_search)

I

id_reedit() (in module mwptoolkit.utils.preprocess_tool.dataset_operator) IN_EQN (mwptoolkit.utils.enum_type.EPT attribute) IN_TNPAD (mwptoolkit.utils.enum_type.EPT attribute) IN_TNUM (mwptoolkit.utils.enum_type.EPT attribute) IN_TPAD (mwptoolkit.utils.enum_type.EPT attribute) IN_TXT (mwptoolkit.utils.enum_type.EPT attribute) Infix (mwptoolkit.utils.enum_type.FixType attribute) infix_to_postfix() (in module mwptoolkit.utils.preprocess_tool.equation_operator) InfixEvaluator (class in mwptoolkit.evaluate.evaluator) init_batches() (mwptoolkit.data.dataloader.abstract_dataloader.AbstractDataLoader method) (mwptoolkit.data.dataloader.multi_equation_dataloader.MultiEquationDataLoader method) (mwptoolkit.data.dataloader.pretrain_dataloader.PretrainDataLoader method) (mwptoolkit.data.dataloader.single_equation_dataloader.SingleEquationDataLoader method) (mwptoolkit.data.dataloader.template_dataloader.TemplateDataLoader method) init_decoder_inputs() (mwptoolkit.model.Seq2Seq.dns.DNS method) (mwptoolkit.model.Seq2Seq.groupatt.GroupATT method) (mwptoolkit.model.Seq2Seq.mathen.MathEN method) (mwptoolkit.model.Seq2Seq.rnnencdec.RNNEncDec method) (mwptoolkit.model.Seq2Seq.rnnvae.RNNVAE method) (mwptoolkit.model.Seq2Seq.transformer.Transformer method)

init_embedding_params() (mwptoolkit.module.Embedder.basic_embedder.BasicEmbedder method) init_encoder_mask() (mwptoolkit.model.Seq2Tree.tsn.TSN method) init_factor() (mwptoolkit.module.Decoder.ept_decoder.DecoderModel method) init_hidden() (mwptoolkit.module.Decoder.rnn_decoder.AttentionalRNNDecoder method) (mwptoolkit.module.Decoder.rnn_decoder.BasicRNNDecoder method) (mwptoolkit.module.Encoder.rnn_encoder.BasicRNNEncoder method) (mwptoolkit.module.Encoder.rnn_encoder.SelfAttentionRNNEncoder method) init_logger() (in module mwptoolkit.utils.logger) init_seed() (in module mwptoolkit.utils.utils) init_seq2seq_decoder_inputs() (mwptoolkit.model.Seq2Tree.trnn.TRNN method) init_soft_target() (mwptoolkit.model.Seq2Tree.tsn.TSN method) init_stacks() (mwptoolkit.module.Decoder.tree_decoder.HMSDecoder method) initialize_fix_constant() (mwptoolkit.module.Encoder.rnn_encoder.SalignedEncoder method) is_equal() (mwptoolkit.utils.data_structure.GoldTree method) is_expression_type (mwptoolkit.module.Decoder.ept_decoder.DecoderModel property) is_float() (mwptoolkit.utils.data_structure.GoldTree method) is_in_rel_quants() (mwptoolkit.utils.data_structure.GoldTree method)

J

joint_fraction() (in module mwptoolkit.utils.preprocess_tool.number_operator)

joint_number() (in module mwptoolkit.utils.preprocess_tool.number_operator) joint_number_() (in module mwptoolkit.utils.preprocess_tool.number_operator)

L

LayerNorm (class in mwptoolkit.module.Layer.graph_layers) (class in mwptoolkit.module.Layer.transformer_layer) lca() (mwptoolkit.utils.data_structure.GoldTree method) leaf_emb() (mwptoolkit.module.Layer.tree_layers.RecursiveNN method) list2str() (mwptoolkit.model.PreTrain.gpt2.GPT2 method) lists2dict() (in module mwptoolkit.utils.utils) load_data() (mwptoolkit.data.dataloader.abstract_dataloader.AbstractDataLoader method) (mwptoolkit.data.dataloader.multi_equation_dataloader.MultiEquationDataLoader method) (mwptoolkit.data.dataloader.pretrain_dataloader.PretrainDataLoader method) (mwptoolkit.data.dataloader.single_equation_dataloader.SingleEquationDataLoader method) (mwptoolkit.data.dataloader.template_dataloader.TemplateDataLoader method) load_from_pretrained() (mwptoolkit.config.configuration.Config class method) (mwptoolkit.data.dataset.abstract_dataset.AbstractDataset class method) (mwptoolkit.data.dataset.dataset_ept.DatasetEPT class method) (mwptoolkit.data.dataset.dataset_hms.DatasetHMS class method) (mwptoolkit.data.dataset.dataset_multiencdec.DatasetMultiEncDec class method) (mwptoolkit.data.dataset.multi_equation_dataset.MultiEquationDataset class method) (mwptoolkit.data.dataset.pretrain_dataset.PretrainDataset class method) (mwptoolkit.data.dataset.single_equation_dataset.SingleEquationDataset class method) (mwptoolkit.data.dataset.template_dataset.TemplateDataset class method)

load_next_batch() (mwptoolkit.data.dataloader.abstract_dataloader.AbstractDataLoader method) (mwptoolkit.data.dataloader.multi_equation_dataloader.MultiEquationDataLoader method) (mwptoolkit.data.dataloader.pretrain_dataloader.PretrainDataLoader method) (mwptoolkit.data.dataloader.single_equation_dataloader.SingleEquationDataLoader method) (mwptoolkit.data.dataloader.template_dataloader.TemplateDataLoader method) LogSoftmax (class in mwptoolkit.module.Decoder.ept_decoder) LSTMBasedTreeDecoder (class in mwptoolkit.module.Decoder.tree_decoder)

M

mask2num() (mwptoolkit.model.Seq2Tree.trnn.TRNN method) mask_forward() (in module mwptoolkit.module.Decoder.ept_decoder) masked_cross_entropy() (in module mwptoolkit.loss.masked_cross_entropy_loss) MaskedCrossEntropyLoss (class in mwptoolkit.loss.masked_cross_entropy_loss) MaskedRelevantScore (class in mwptoolkit.module.Attention.seq_attention) MaskSymbol (class in mwptoolkit.utils.enum_type) math23k (mwptoolkit.utils.enum_type.DatasetName attribute) MathEN (class in mwptoolkit.model.Seq2Seq.mathen) mawps (mwptoolkit.utils.enum_type.DatasetName attribute) mawps_asdiv_a_svamp (mwptoolkit.utils.enum_type.DatasetName attribute) mawps_single (mwptoolkit.utils.enum_type.DatasetName attribute) MeanAggregator (class in mwptoolkit.module.Layer.graph_layers) MEM_MAX (mwptoolkit.utils.enum_type.EPT attribute) MEM_PREFIX (mwptoolkit.utils.enum_type.EPT attribute) Merge (class in mwptoolkit.module.Layer.tree_layers) merge() (mwptoolkit.module.Layer.tree_layers.TreeEmbeddingModel method) MODEL_EXPR_PTR_TRANS (mwptoolkit.utils.enum_type.EPT attribute) MODEL_EXPR_TRANS (mwptoolkit.utils.enum_type.EPT attribute) model_test() (mwptoolkit.model.Graph2Tree.graph2tree.Graph2Tree method) (mwptoolkit.model.Graph2Tree.multiencdec.MultiEncDec method) (mwptoolkit.model.PreTrain.bertgen.BERTGen method) (mwptoolkit.model.PreTrain.gpt2.GPT2 method) (mwptoolkit.model.PreTrain.robertagen.RobertaGen method) (mwptoolkit.model.Seq2Seq.dns.DNS method) (mwptoolkit.model.Seq2Seq.ept.EPT method) (mwptoolkit.model.Seq2Seq.groupatt.GroupATT method) (mwptoolkit.model.Seq2Seq.mathen.MathEN method) (mwptoolkit.model.Seq2Seq.rnnencdec.RNNEncDec method) (mwptoolkit.model.Seq2Seq.rnnvae.RNNVAE method) (mwptoolkit.model.Seq2Seq.saligned.Saligned method) (mwptoolkit.model.Seq2Seq.transformer.Transformer method) (mwptoolkit.model.Seq2Tree.berttd.BertTD method) (mwptoolkit.model.Seq2Tree.gts.GTS method) (mwptoolkit.model.Seq2Tree.mwpbert.MWPBert method) (mwptoolkit.model.Seq2Tree.sausolver.SAUSolver method) (mwptoolkit.model.Seq2Tree.treelstm.TreeLSTM method) (mwptoolkit.model.Seq2Tree.trnn.TRNN method) (mwptoolkit.model.Seq2Tree.tsn.TSN method) MODEL_VANILLA_TRANS (mwptoolkit.utils.enum_type.EPT attribute) module mwptoolkit.config.configuration mwptoolkit.data.dataloader.abstract_dataloader mwptoolkit.data.dataloader.dataloader_ept mwptoolkit.data.dataloader.dataloader_hms mwptoolkit.data.dataloader.dataloader_multiencdec mwptoolkit.data.dataloader.multi_equation_dataloader mwptoolkit.data.dataloader.pretrain_dataloader mwptoolkit.data.dataloader.single_equation_dataloader mwptoolkit.data.dataloader.template_dataloader mwptoolkit.data.dataset.abstract_dataset mwptoolkit.data.dataset.dataset_ept mwptoolkit.data.dataset.dataset_hms mwptoolkit.data.dataset.dataset_multiencdec mwptoolkit.data.dataset.multi_equation_dataset mwptoolkit.data.dataset.pretrain_dataset mwptoolkit.data.dataset.single_equation_dataset mwptoolkit.data.dataset.template_dataset mwptoolkit.data.utils mwptoolkit.evaluate.evaluator mwptoolkit.hyper_search mwptoolkit.loss.abstract_loss mwptoolkit.loss.binary_cross_entropy_loss mwptoolkit.loss.cross_entropy_loss mwptoolkit.loss.masked_cross_entropy_loss mwptoolkit.loss.mse_loss mwptoolkit.loss.nll_loss mwptoolkit.loss.smoothed_cross_entropy_loss mwptoolkit.model.Graph2Tree.graph2tree mwptoolkit.model.Graph2Tree.multiencdec mwptoolkit.model.PreTrain.bertgen mwptoolkit.model.PreTrain.gpt2 mwptoolkit.model.PreTrain.robertagen mwptoolkit.model.Seq2Seq.dns mwptoolkit.model.Seq2Seq.ept mwptoolkit.model.Seq2Seq.groupatt mwptoolkit.model.Seq2Seq.mathen mwptoolkit.model.Seq2Seq.rnnencdec mwptoolkit.model.Seq2Seq.rnnvae mwptoolkit.model.Seq2Seq.saligned mwptoolkit.model.Seq2Seq.transformer mwptoolkit.model.Seq2Tree.berttd mwptoolkit.model.Seq2Tree.gts mwptoolkit.model.Seq2Tree.mwpbert mwptoolkit.model.Seq2Tree.sausolver mwptoolkit.model.Seq2Tree.treelstm mwptoolkit.model.Seq2Tree.trnn mwptoolkit.model.Seq2Tree.tsn mwptoolkit.module.Attention.group_attention mwptoolkit.module.Attention.multi_head_attention mwptoolkit.module.Attention.self_attention mwptoolkit.module.Attention.seq_attention mwptoolkit.module.Attention.tree_attention mwptoolkit.module.Decoder.ept_decoder mwptoolkit.module.Decoder.rnn_decoder mwptoolkit.module.Decoder.transformer_decoder mwptoolkit.module.Decoder.tree_decoder mwptoolkit.module.Embedder.basic_embedder mwptoolkit.module.Embedder.bert_embedder mwptoolkit.module.Embedder.position_embedder mwptoolkit.module.Embedder.roberta_embedder mwptoolkit.module.Encoder.graph_based_encoder mwptoolkit.module.Encoder.rnn_encoder mwptoolkit.module.Encoder.transformer_encoder mwptoolkit.module.Environment.stack_machine mwptoolkit.module.Graph.gcn mwptoolkit.module.Graph.graph_module mwptoolkit.module.Layer.graph_layers mwptoolkit.module.Layer.layers mwptoolkit.module.Layer.transformer_layer mwptoolkit.module.Layer.tree_layers mwptoolkit.module.Strategy.beam_search mwptoolkit.module.Strategy.greedy mwptoolkit.module.Strategy.sampling mwptoolkit.quick_start mwptoolkit.trainer.abstract_trainer mwptoolkit.trainer.supervised_trainer mwptoolkit.trainer.template_trainer mwptoolkit.utils.data_structure mwptoolkit.utils.enum_type mwptoolkit.utils.logger mwptoolkit.utils.preprocess_tool.dataset_operator mwptoolkit.utils.preprocess_tool.equation_operator mwptoolkit.utils.preprocess_tool.number_operator mwptoolkit.utils.preprocess_tool.number_transfer mwptoolkit.utils.preprocess_tool.sentence_operator mwptoolkit.utils.utils mse_loss() (mwptoolkit.model.Seq2Tree.sausolver.SAUSolver method) MSELoss (class in mwptoolkit.loss.mse_loss) Multi (mwptoolkit.utils.enum_type.Operators attribute) MultiEncDec (class in mwptoolkit.model.Graph2Tree.multiencdec) MultiEncDecEvaluator (class in mwptoolkit.evaluate.evaluator) MultiEncDecTrainer (class in mwptoolkit.trainer.supervised_trainer) MultiEquation (mwptoolkit.utils.enum_type.TaskType attribute) MultiEquationDataLoader (class in mwptoolkit.data.dataloader.multi_equation_dataloader) MultiEquationDataset (class in mwptoolkit.data.dataset.multi_equation_dataset) MultiHeadAttention (class in mwptoolkit.module.Attention.multi_head_attention) MULTIPLES (mwptoolkit.utils.enum_type.EPT attribute) MultiWayTree (mwptoolkit.utils.enum_type.FixType attribute) MultiWayTreeEvaluator (class in mwptoolkit.evaluate.evaluator) MWPBert (class in mwptoolkit.model.Seq2Tree.mwpbert) MWPBertTrainer (class in mwptoolkit.trainer.supervised_trainer) mwptoolkit.config.configuration module mwptoolkit.data.dataloader.abstract_dataloader module mwptoolkit.data.dataloader.dataloader_ept module mwptoolkit.data.dataloader.dataloader_hms module mwptoolkit.data.dataloader.dataloader_multiencdec module mwptoolkit.data.dataloader.multi_equation_dataloader module mwptoolkit.data.dataloader.pretrain_dataloader module mwptoolkit.data.dataloader.single_equation_dataloader module

mwptoolkit.data.dataloader.template_dataloader module mwptoolkit.data.dataset.abstract_dataset module mwptoolkit.data.dataset.dataset_ept module mwptoolkit.data.dataset.dataset_hms module mwptoolkit.data.dataset.dataset_multiencdec module mwptoolkit.data.dataset.multi_equation_dataset module mwptoolkit.data.dataset.pretrain_dataset module mwptoolkit.data.dataset.single_equation_dataset module mwptoolkit.data.dataset.template_dataset module mwptoolkit.data.utils module mwptoolkit.evaluate.evaluator module mwptoolkit.hyper_search module mwptoolkit.loss.abstract_loss module mwptoolkit.loss.binary_cross_entropy_loss module mwptoolkit.loss.cross_entropy_loss module mwptoolkit.loss.masked_cross_entropy_loss module mwptoolkit.loss.mse_loss module mwptoolkit.loss.nll_loss module mwptoolkit.loss.smoothed_cross_entropy_loss module mwptoolkit.model.Graph2Tree.graph2tree module mwptoolkit.model.Graph2Tree.multiencdec module mwptoolkit.model.PreTrain.bertgen module mwptoolkit.model.PreTrain.gpt2 module mwptoolkit.model.PreTrain.robertagen module mwptoolkit.model.Seq2Seq.dns module mwptoolkit.model.Seq2Seq.ept module mwptoolkit.model.Seq2Seq.groupatt module mwptoolkit.model.Seq2Seq.mathen module mwptoolkit.model.Seq2Seq.rnnencdec module mwptoolkit.model.Seq2Seq.rnnvae module mwptoolkit.model.Seq2Seq.saligned module mwptoolkit.model.Seq2Seq.transformer module mwptoolkit.model.Seq2Tree.berttd module mwptoolkit.model.Seq2Tree.gts module mwptoolkit.model.Seq2Tree.mwpbert module mwptoolkit.model.Seq2Tree.sausolver module mwptoolkit.model.Seq2Tree.treelstm module mwptoolkit.model.Seq2Tree.trnn module mwptoolkit.model.Seq2Tree.tsn module mwptoolkit.module.Attention.group_attention module mwptoolkit.module.Attention.multi_head_attention module mwptoolkit.module.Attention.self_attention module mwptoolkit.module.Attention.seq_attention module mwptoolkit.module.Attention.tree_attention module mwptoolkit.module.Decoder.ept_decoder module mwptoolkit.module.Decoder.rnn_decoder module mwptoolkit.module.Decoder.transformer_decoder module mwptoolkit.module.Decoder.tree_decoder module mwptoolkit.module.Embedder.basic_embedder module mwptoolkit.module.Embedder.bert_embedder module mwptoolkit.module.Embedder.position_embedder module mwptoolkit.module.Embedder.roberta_embedder module mwptoolkit.module.Encoder.graph_based_encoder module mwptoolkit.module.Encoder.rnn_encoder module mwptoolkit.module.Encoder.transformer_encoder module mwptoolkit.module.Environment.stack_machine module mwptoolkit.module.Graph.gcn module mwptoolkit.module.Graph.graph_module module mwptoolkit.module.Layer.graph_layers module mwptoolkit.module.Layer.layers module mwptoolkit.module.Layer.transformer_layer module mwptoolkit.module.Layer.tree_layers module mwptoolkit.module.Strategy.beam_search module mwptoolkit.module.Strategy.greedy module mwptoolkit.module.Strategy.sampling module mwptoolkit.quick_start module mwptoolkit.trainer.abstract_trainer module mwptoolkit.trainer.supervised_trainer module mwptoolkit.trainer.template_trainer module mwptoolkit.utils.data_structure module mwptoolkit.utils.enum_type module mwptoolkit.utils.logger module mwptoolkit.utils.preprocess_tool.dataset_operator module mwptoolkit.utils.preprocess_tool.equation_operator module mwptoolkit.utils.preprocess_tool.number_operator module mwptoolkit.utils.preprocess_tool.number_transfer module mwptoolkit.utils.preprocess_tool.sentence_operator module mwptoolkit.utils.utils module

N

NEG_INF (mwptoolkit.utils.enum_type.EPT attribute) NLLLoss (class in mwptoolkit.loss.nll_loss) Node (class in mwptoolkit.module.Layer.tree_layers) (class in mwptoolkit.utils.data_structure) NodeEmbeddingLayer (class in mwptoolkit.module.Layer.tree_layers) NodeEmbeddingNode (class in mwptoolkit.module.Layer.tree_layers) NodeGenerater (class in mwptoolkit.module.Layer.tree_layers) NON_TOKEN (mwptoolkit.utils.enum_type.SpecialTokens attribute) Nonfix (mwptoolkit.utils.enum_type.FixType attribute) normalize() (mwptoolkit.module.Graph.graph_module.Graph_Module method) (mwptoolkit.module.Graph.graph_module.Num_Graph_Module method) (mwptoolkit.module.Graph.graph_module.Parse_Graph_Module method) NUM (mwptoolkit.utils.enum_type.MaskSymbol attribute) (mwptoolkit.utils.enum_type.NumMask attribute) Num_Graph_Module (class in mwptoolkit.module.Graph.graph_module) num_heads (mwptoolkit.module.Attention.multi_head_attention.EPTMultiHeadAttentionWeights property) NUM_MAX (mwptoolkit.utils.enum_type.EPT attribute) num_order_processed() (mwptoolkit.data.dataloader.dataloader_multiencdec.DataLoaderMultiEncDec method) NUM_PREFIX (mwptoolkit.utils.enum_type.EPT attribute)

NUM_TOKEN (mwptoolkit.utils.enum_type.EPT attribute) num_transfer_alg514() (in module mwptoolkit.utils.preprocess_tool.number_transfer) num_transfer_draw() (in module mwptoolkit.utils.preprocess_tool.number_transfer) num_transfer_hmwp() (in module mwptoolkit.utils.preprocess_tool.number_transfer) num_transfer_multi() (in module mwptoolkit.utils.preprocess_tool.number_transfer) number (mwptoolkit.utils.enum_type.MaskSymbol attribute) (mwptoolkit.utils.enum_type.NumMask attribute) NUMBER_AND_FRACTION_PATTERN (mwptoolkit.utils.enum_type.EPT attribute) NUMBER_PATTERN (mwptoolkit.utils.enum_type.EPT attribute) NUMBER_READINGS (mwptoolkit.utils.enum_type.EPT attribute) number_transfer() (in module mwptoolkit.utils.preprocess_tool.number_transfer) number_transfer_ape200k() (in module mwptoolkit.utils.preprocess_tool.number_transfer) number_transfer_asdiv_a() (in module mwptoolkit.utils.preprocess_tool.number_transfer) number_transfer_math23k() (in module mwptoolkit.utils.preprocess_tool.number_transfer) number_transfer_mawps() (in module mwptoolkit.utils.preprocess_tool.number_transfer) number_transfer_mawps_single() (in module mwptoolkit.utils.preprocess_tool.number_transfer) number_transfer_single() (in module mwptoolkit.utils.preprocess_tool.number_transfer) number_transfer_svamp() (in module mwptoolkit.utils.preprocess_tool.number_transfer) NumEncoder (class in mwptoolkit.module.Encoder.graph_based_encoder) NumMask (class in mwptoolkit.utils.enum_type)

O

OpDecoderModel (class in mwptoolkit.module.Decoder.ept_decoder) operand_norm (mwptoolkit.module.Decoder.ept_decoder.ExpressionDecoderModel attribute) operand_out (mwptoolkit.module.Decoder.ept_decoder.ExpressionPointerTransformer attribute) (mwptoolkit.module.Decoder.ept_decoder.ExpressionTransformer attribute) operand_source_embedding (mwptoolkit.module.Decoder.ept_decoder.ExpressionDecoderModel attribute) operand_source_factor (mwptoolkit.module.Decoder.ept_decoder.ExpressionDecoderModel attribute) operand_word_embedding (mwptoolkit.module.Decoder.ept_decoder.ExpressionTransformer attribute) OPERATIONS (class in mwptoolkit.module.Environment.stack_machine)

operator_mask() (in module mwptoolkit.utils.preprocess_tool.equation_operator) operator_mask_process() (mwptoolkit.data.dataset.abstract_dataset.AbstractDataset method) OPERATOR_PRECEDENCE (mwptoolkit.utils.enum_type.EPT attribute) Operators (class in mwptoolkit.utils.enum_type) OPERATORS (mwptoolkit.utils.enum_type.EPT attribute) OPT_TOKEN (mwptoolkit.utils.enum_type.SpecialTokens attribute) orig_infix_to_postfix() (in module mwptoolkit.utils.preprocess_tool.equation_operator) out_expression_expr() (mwptoolkit.model.Seq2Seq.ept.EPT method) out_expression_op() (mwptoolkit.model.Seq2Seq.ept.EPT method)

P

pad_and_cat() (mwptoolkit.module.Decoder.rnn_decoder.SalignedDecoder method) PAD_ID (mwptoolkit.utils.enum_type.EPT attribute) PAD_TOKEN (mwptoolkit.utils.enum_type.SpecialTokens attribute) param_search() (mwptoolkit.trainer.abstract_trainer.AbstractTrainer method) (mwptoolkit.trainer.supervised_trainer.EPTTrainer method) (mwptoolkit.trainer.supervised_trainer.GTSTrainer method) (mwptoolkit.trainer.supervised_trainer.SupervisedTrainer method) (mwptoolkit.trainer.supervised_trainer.TreeLSTMTrainer method) (mwptoolkit.trainer.supervised_trainer.TRNNTrainer method) parameters_to_dict() (mwptoolkit.data.dataset.abstract_dataset.AbstractDataset method) Parse_Graph_Module (class in mwptoolkit.module.Graph.graph_module) play_one() (mwptoolkit.module.Layer.tree_layers.DQN method) PLURAL_FORMS (mwptoolkit.utils.enum_type.EPT attribute) pos_factor (mwptoolkit.module.Decoder.ept_decoder.OpDecoderModel attribute) POS_INF (mwptoolkit.utils.enum_type.EPT attribute) PositionalEncoding (class in mwptoolkit.module.Embedder.position_embedder) PositionEmbedder (class in mwptoolkit.module.Embedder.position_embedder) PositionEmbedder_x (class in mwptoolkit.module.Embedder.position_embedder) PositionwiseFeedForward (class in mwptoolkit.module.Layer.graph_layers) (class in mwptoolkit.module.Layer.transformer_layer) Postfix (mwptoolkit.utils.enum_type.FixType attribute) postfix_parser() (in module mwptoolkit.utils.preprocess_tool.equation_operator) postfix_result() (mwptoolkit.evaluate.evaluator.MultiEncDecEvaluator method) postfix_result_multi() (mwptoolkit.evaluate.evaluator.MultiEncDecEvaluator method) PostfixEvaluator (class in mwptoolkit.evaluate.evaluator) predict() (mwptoolkit.model.Graph2Tree.graph2tree.Graph2Tree method) (mwptoolkit.model.Graph2Tree.multiencdec.MultiEncDec method) (mwptoolkit.model.PreTrain.bertgen.BERTGen method) (mwptoolkit.model.PreTrain.gpt2.GPT2 method) (mwptoolkit.model.PreTrain.robertagen.RobertaGen method) (mwptoolkit.model.Seq2Seq.dns.DNS method) (mwptoolkit.model.Seq2Seq.ept.EPT method) (mwptoolkit.model.Seq2Seq.groupatt.GroupATT method) (mwptoolkit.model.Seq2Seq.mathen.MathEN method) (mwptoolkit.model.Seq2Seq.rnnencdec.RNNEncDec method) (mwptoolkit.model.Seq2Seq.rnnvae.RNNVAE method) (mwptoolkit.model.Seq2Seq.saligned.Saligned method) (mwptoolkit.model.Seq2Seq.transformer.Transformer method) (mwptoolkit.model.Seq2Tree.berttd.BertTD method) (mwptoolkit.model.Seq2Tree.gts.GTS method) (mwptoolkit.model.Seq2Tree.mwpbert.MWPBert method) (mwptoolkit.model.Seq2Tree.sausolver.SAUSolver method) (mwptoolkit.model.Seq2Tree.treelstm.TreeLSTM method) (mwptoolkit.model.Seq2Tree.trnn.TRNN method) (mwptoolkit.model.Seq2Tree.tsn.TSN method)

Prediction (class in mwptoolkit.module.Layer.tree_layers) PredictModel (class in mwptoolkit.module.Decoder.tree_decoder) Prefix (mwptoolkit.utils.enum_type.FixType attribute) prefix2tree() (mwptoolkit.utils.data_structure.PrefixTree method) prefix_result() (mwptoolkit.evaluate.evaluator.MultiEncDecEvaluator method) prefix_result_multi() (mwptoolkit.evaluate.evaluator.MultiEncDecEvaluator method) PrefixEvaluator (class in mwptoolkit.evaluate.evaluator) PrefixTree (class in mwptoolkit.utils.data_structure) PREP_KEY_ANS (mwptoolkit.utils.enum_type.EPT attribute) PREP_KEY_EQN (mwptoolkit.utils.enum_type.EPT attribute) PREP_KEY_MEM (mwptoolkit.utils.enum_type.EPT attribute) preprocess_ept_dataset_() (in module mwptoolkit.utils.preprocess_tool.dataset_operator) PretrainDataLoader (class in mwptoolkit.data.dataloader.pretrain_dataloader) PretrainDataset (class in mwptoolkit.data.dataset.pretrain_dataset) PretrainSeq2SeqTrainer (class in mwptoolkit.trainer.supervised_trainer) PretrainTRNNTrainer (class in mwptoolkit.trainer.supervised_trainer) problem_level_forword() (mwptoolkit.module.Encoder.rnn_encoder.HWCPEncoder method) process_gap_encoder_decoder() (mwptoolkit.model.Seq2Seq.groupatt.GroupATT method) push() (mwptoolkit.module.Environment.stack_machine.StackMachine method)

Q

query() (mwptoolkit.utils.data_structure.GoldTree method)

R

read_ape200k_source() (in module mwptoolkit.utils.utils) read_json_data() (in module mwptoolkit.utils.utils) read_math23k_source() (in module mwptoolkit.utils.utils) read_pos_from_file() (mwptoolkit.data.dataset.dataset_multiencdec.DatasetMultiEncDec method) RecurCell() (mwptoolkit.module.Layer.tree_layers.RecursiveNN method) RecursiveNN (class in mwptoolkit.module.Layer.tree_layers) refine_formula_as_prefix() (in module mwptoolkit.utils.preprocess_tool.dataset_operator) RelevantScore (class in mwptoolkit.module.Attention.seq_attention) replace_masked_values() (in module mwptoolkit.module.Encoder.graph_based_encoder) required_field (mwptoolkit.module.Decoder.ept_decoder.DecoderModel property) (mwptoolkit.module.Decoder.ept_decoder.ExpressionPointerTransformer property) (mwptoolkit.module.Decoder.ept_decoder.ExpressionTransformer property) (mwptoolkit.module.Decoder.ept_decoder.VanillaOpTransformer property) reset() (mwptoolkit.loss.abstract_loss.AbstractLoss method) reset_dataset() (mwptoolkit.data.dataset.abstract_dataset.AbstractDataset method) reset_parameters() (mwptoolkit.module.Layer.graph_layers.GraphConvolution method) (mwptoolkit.module.Layer.transformer_layer.TransformerLayer method) result() (mwptoolkit.evaluate.evaluator.AbstractEvaluator method) (mwptoolkit.evaluate.evaluator.InfixEvaluator method) (mwptoolkit.evaluate.evaluator.MultiEncDecEvaluator method) (mwptoolkit.evaluate.evaluator.MultiWayTreeEvaluator method) (mwptoolkit.evaluate.evaluator.PostfixEvaluator method) (mwptoolkit.evaluate.evaluator.PrefixEvaluator method)

result_multi() (mwptoolkit.evaluate.evaluator.AbstractEvaluator method) (mwptoolkit.evaluate.evaluator.InfixEvaluator method) (mwptoolkit.evaluate.evaluator.MultiEncDecEvaluator method) (mwptoolkit.evaluate.evaluator.MultiWayTreeEvaluator method) (mwptoolkit.evaluate.evaluator.PostfixEvaluator method) (mwptoolkit.evaluate.evaluator.PrefixEvaluator method) RNNBasedTreeDecoder (class in mwptoolkit.module.Decoder.tree_decoder) RNNEncDec (class in mwptoolkit.model.Seq2Seq.rnnencdec) RNNVAE (class in mwptoolkit.model.Seq2Seq.rnnvae) RobertaEmbedder (class in mwptoolkit.module.Embedder.roberta_embedder) RobertaGen (class in mwptoolkit.model.PreTrain.robertagen) rule1_filter() (mwptoolkit.model.Seq2Seq.dns.DNS method) rule2_filter() (mwptoolkit.model.Seq2Seq.dns.DNS method) rule3_filter() (mwptoolkit.model.Seq2Seq.dns.DNS method) rule4_filter() (mwptoolkit.model.Seq2Seq.dns.DNS method) rule5_filter() (mwptoolkit.model.Seq2Seq.dns.DNS method) rule_filter_() (mwptoolkit.model.Seq2Seq.dns.DNS method) run() (mwptoolkit.evaluate.evaluator.Solver method) run_toolkit() (in module mwptoolkit.quick_start)

S

Saligned (class in mwptoolkit.model.Seq2Seq.saligned) SalignedDecoder (class in mwptoolkit.module.Decoder.rnn_decoder) SalignedEncoder (class in mwptoolkit.module.Encoder.rnn_encoder) SalignedTrainer (class in mwptoolkit.trainer.supervised_trainer) SARTreeDecoder (class in mwptoolkit.module.Decoder.tree_decoder) SAUSolver (class in mwptoolkit.model.Seq2Tree.sausolver) SAUSolverTrainer (class in mwptoolkit.trainer.supervised_trainer) save_config() (mwptoolkit.config.configuration.Config method) save_dataset() (mwptoolkit.data.dataset.abstract_dataset.AbstractDataset method) (mwptoolkit.data.dataset.dataset_ept.DatasetEPT method) (mwptoolkit.data.dataset.dataset_hms.DatasetHMS method) (mwptoolkit.data.dataset.dataset_multiencdec.DatasetMultiEncDec method) (mwptoolkit.data.dataset.multi_equation_dataset.MultiEquationDataset method) (mwptoolkit.data.dataset.pretrain_dataset.PretrainDataset method) (mwptoolkit.data.dataset.single_equation_dataset.SingleEquationDataset method) (mwptoolkit.data.dataset.template_dataset.TemplateDataset method) Score (class in mwptoolkit.module.Layer.tree_layers) score_pn() (mwptoolkit.module.Decoder.tree_decoder.PredictModel method) ScoreModel (class in mwptoolkit.module.Layer.tree_layers) seg_and_tag_ape200k() (in module mwptoolkit.utils.preprocess_tool.number_transfer) seg_and_tag_asdiv_a() (in module mwptoolkit.utils.preprocess_tool.number_transfer) seg_and_tag_hmwp() (in module mwptoolkit.utils.preprocess_tool.number_transfer) seg_and_tag_math23k() (in module mwptoolkit.utils.preprocess_tool.number_transfer) seg_and_tag_mawps() (in module mwptoolkit.utils.preprocess_tool.number_transfer) seg_and_tag_mawps_single() (in module mwptoolkit.utils.preprocess_tool.number_transfer) seg_and_tag_multi() (in module mwptoolkit.utils.preprocess_tool.number_transfer) seg_and_tag_single() (in module mwptoolkit.utils.preprocess_tool.number_transfer) seg_and_tag_svamp() (in module mwptoolkit.utils.preprocess_tool.number_transfer) SelfAttention (class in mwptoolkit.module.Attention.self_attention) SelfAttentionMask (class in mwptoolkit.module.Attention.self_attention) SelfAttentionRNNEncoder (class in mwptoolkit.module.Encoder.rnn_encoder) SemanticAlignmentModule (class in mwptoolkit.module.Layer.tree_layers) Semantically_Aligned_Regularization() (mwptoolkit.module.Decoder.tree_decoder.SARTreeDecoder method) sentence2tree() (mwptoolkit.utils.data_structure.DependencyTree method) seq2seq_calculate_loss() (mwptoolkit.model.Seq2Tree.trnn.TRNN method) seq2seq_decoder_forward() (mwptoolkit.model.Seq2Tree.trnn.TRNN method) seq2seq_encoder_forward() (mwptoolkit.model.Seq2Tree.trnn.TRNN method) seq2seq_forward() (mwptoolkit.model.Seq2Tree.trnn.TRNN method) seq2seq_generate_t() (mwptoolkit.model.Seq2Tree.trnn.TRNN method) seq2seq_generate_without_t() (mwptoolkit.model.Seq2Tree.trnn.TRNN method) SEQ_END_EQN (mwptoolkit.utils.enum_type.EPT attribute) SEQ_END_EQN_ID (mwptoolkit.utils.enum_type.EPT attribute) SEQ_EQ_SGN_ID (mwptoolkit.utils.enum_type.EPT attribute) SEQ_GEN_NUM_ID (mwptoolkit.utils.enum_type.EPT attribute) SEQ_GEN_VAR_ID (mwptoolkit.utils.enum_type.EPT attribute) SEQ_NEW_EQN (mwptoolkit.utils.enum_type.EPT attribute) SEQ_NEW_EQN_ID (mwptoolkit.utils.enum_type.EPT attribute) SEQ_PTR_NUM (mwptoolkit.utils.enum_type.EPT attribute) SEQ_PTR_NUM_ID (mwptoolkit.utils.enum_type.EPT attribute) SEQ_PTR_TOKENS (mwptoolkit.utils.enum_type.EPT attribute)

SEQ_PTR_VAR (mwptoolkit.utils.enum_type.EPT attribute) SEQ_PTR_VAR_ID (mwptoolkit.utils.enum_type.EPT attribute) SEQ_TOKENS (mwptoolkit.utils.enum_type.EPT attribute) SEQ_UNK_TOK (mwptoolkit.utils.enum_type.EPT attribute) SEQ_UNK_TOK_ID (mwptoolkit.utils.enum_type.EPT attribute) SeqAttention (class in mwptoolkit.module.Attention.seq_attention) sequence_mask() (in module mwptoolkit.loss.masked_cross_entropy_loss) set_left_node() (mwptoolkit.module.Layer.tree_layers.Node method) (mwptoolkit.utils.data_structure.Node method) set_right_node() (mwptoolkit.module.Layer.tree_layers.Node method) (mwptoolkit.utils.data_structure.Node method) shared_decoder_layer (mwptoolkit.module.Decoder.ept_decoder.ExpressionDecoderModel attribute) shift_target() (mwptoolkit.model.Seq2Seq.ept.EPT method) Single (mwptoolkit.utils.enum_type.Operators attribute) SingleEquation (mwptoolkit.utils.enum_type.TaskType attribute) SingleEquationDataLoader (class in mwptoolkit.data.dataloader.single_equation_dataloader) SingleEquationDataset (class in mwptoolkit.data.dataset.single_equation_dataset) SmoothCrossEntropyLoss (class in mwptoolkit.loss.smoothed_cross_entropy_loss) SmoothedCrossEntropyLoss (class in mwptoolkit.loss.smoothed_cross_entropy_loss) soft_cross_entropy_loss() (in module mwptoolkit.model.Seq2Tree.tsn) soft_target_loss() (in module mwptoolkit.model.Seq2Tree.tsn) softmax (mwptoolkit.module.Decoder.ept_decoder.VanillaOpTransformer attribute) Solver (class in mwptoolkit.evaluate.evaluator) SOS_TOKEN (mwptoolkit.utils.enum_type.SpecialTokens attribute) span_level_deprel_tree_to_file() (in module mwptoolkit.utils.preprocess_tool.sentence_operator) SpecialTokens (class in mwptoolkit.utils.enum_type) SPIECE_UNDERLINE (mwptoolkit.utils.enum_type.EPT attribute) split_number() (in module mwptoolkit.utils.preprocess_tool.number_operator) split_sentence() (in module mwptoolkit.utils.preprocess_tool.sentence_operator) Squeeze (class in mwptoolkit.module.Decoder.ept_decoder) src_to_mask() (in module mwptoolkit.module.Attention.group_attention) (mwptoolkit.module.Attention.group_attention.GroupAttention method) StackMachine (class in mwptoolkit.module.Environment.stack_machine) step() (mwptoolkit.module.Strategy.beam_search.Beam_Search_Hypothesis method) stop() (mwptoolkit.module.Strategy.beam_search.Beam_Search_Hypothesis method) str2float() (in module mwptoolkit.utils.utils) student_calculate_loss() (mwptoolkit.model.Seq2Tree.tsn.TSN method) student_net_1_decoder_forward() (mwptoolkit.model.Seq2Tree.tsn.TSN method) student_net_2_decoder_forward() (mwptoolkit.model.Seq2Tree.tsn.TSN method) student_net_decoder_forward() (mwptoolkit.model.Seq2Tree.tsn.TSN method) student_net_encoder_forward() (mwptoolkit.model.Seq2Tree.tsn.TSN method) student_net_forward() (mwptoolkit.model.Seq2Tree.tsn.TSN method) student_test() (mwptoolkit.model.Seq2Tree.tsn.TSN method) SublayerConnection (class in mwptoolkit.module.Layer.transformer_layer) Submodule_types() (in module mwptoolkit.model.Seq2Seq.ept) SubTreeMerger (class in mwptoolkit.module.Layer.tree_layers) SupervisedTrainer (class in mwptoolkit.trainer.supervised_trainer) SupervisingMode (class in mwptoolkit.utils.enum_type) SVAMP (mwptoolkit.utils.enum_type.DatasetName attribute) symbol2idx() (mwptoolkit.model.Seq2Tree.trnn.TRNN method)

T

TaskType (class in mwptoolkit.utils.enum_type) teacher_calculate_loss() (mwptoolkit.model.Seq2Tree.tsn.TSN method) teacher_net_decoder_forward() (mwptoolkit.model.Seq2Tree.tsn.TSN method) teacher_net_encoder_forward() (mwptoolkit.model.Seq2Tree.tsn.TSN method) teacher_net_forward() (mwptoolkit.model.Seq2Tree.tsn.TSN method) teacher_test() (mwptoolkit.model.Seq2Tree.tsn.TSN method) template2tree() (mwptoolkit.model.Seq2Tree.trnn.TRNN method) TemplateDataLoader (class in mwptoolkit.data.dataloader.template_dataloader) TemplateDataset (class in mwptoolkit.data.dataset.template_dataset) TemplateTrainer (class in mwptoolkit.trainer.template_trainer) Test (mwptoolkit.utils.enum_type.DatasetType attribute) test() (mwptoolkit.module.Layer.tree_layers.RecursiveNN method) (mwptoolkit.trainer.abstract_trainer.AbstractTrainer method) (mwptoolkit.trainer.supervised_trainer.EPTTrainer method) (mwptoolkit.trainer.supervised_trainer.GTSTrainer method) (mwptoolkit.trainer.supervised_trainer.SalignedTrainer method) (mwptoolkit.trainer.supervised_trainer.SupervisedTrainer method) (mwptoolkit.trainer.supervised_trainer.TreeLSTMTrainer method) (mwptoolkit.trainer.supervised_trainer.TRNNTrainer method) (mwptoolkit.trainer.supervised_trainer.TSNTrainer method) (mwptoolkit.trainer.template_trainer.TemplateTrainer method) test_traverse() (mwptoolkit.module.Layer.tree_layers.RecursiveNN method) test_with_cross_validation() (in module mwptoolkit.quick_start) test_with_train_valid_test_split() (in module mwptoolkit.quick_start) time_since() (in module mwptoolkit.utils.utils) to_dict() (mwptoolkit.config.configuration.Config method) to_list() (mwptoolkit.utils.data_structure.Tree method) to_string() (mwptoolkit.utils.data_structure.Tree method) token_resize() (mwptoolkit.module.Embedder.bert_embedder.BertEmbedder method) (mwptoolkit.module.Embedder.roberta_embedder.RobertaEmbedder method) (mwptoolkit.module.Encoder.transformer_encoder.BertEncoder method) TOP_LEVEL_CLASSES (mwptoolkit.utils.enum_type.EPT attribute) topk_sampling() (in module mwptoolkit.module.Strategy.sampling) Train (mwptoolkit.utils.enum_type.DatasetType attribute) train_cross_validation() (in module mwptoolkit.quick_start) train_process() (in module mwptoolkit.hyper_search) train_tree() (mwptoolkit.model.Seq2Tree.sausolver.SAUSolver method) train_with_cross_validation() (in module mwptoolkit.quick_start) train_with_train_valid_test_split() (in module mwptoolkit.quick_start) training (mwptoolkit.loss.smoothed_cross_entropy_loss.SmoothedCrossEntropyLoss attribute) (mwptoolkit.model.Graph2Tree.graph2tree.Graph2Tree attribute) (mwptoolkit.model.Graph2Tree.multiencdec.MultiEncDec attribute) (mwptoolkit.model.PreTrain.bertgen.BERTGen attribute) (mwptoolkit.model.PreTrain.gpt2.GPT2 attribute) (mwptoolkit.model.PreTrain.robertagen.RobertaGen attribute) (mwptoolkit.model.Seq2Seq.dns.DNS attribute) (mwptoolkit.model.Seq2Seq.ept.EPT attribute) (mwptoolkit.model.Seq2Seq.groupatt.GroupATT attribute) (mwptoolkit.model.Seq2Seq.mathen.MathEN attribute) (mwptoolkit.model.Seq2Seq.rnnencdec.RNNEncDec attribute) (mwptoolkit.model.Seq2Seq.rnnvae.RNNVAE attribute) (mwptoolkit.model.Seq2Seq.saligned.Saligned attribute) (mwptoolkit.model.Seq2Seq.transformer.Transformer attribute) (mwptoolkit.model.Seq2Tree.berttd.BertTD attribute) (mwptoolkit.model.Seq2Tree.gts.GTS attribute) (mwptoolkit.model.Seq2Tree.mwpbert.MWPBert attribute) (mwptoolkit.model.Seq2Tree.sausolver.SAUSolver attribute) (mwptoolkit.model.Seq2Tree.treelstm.TreeLSTM attribute) (mwptoolkit.model.Seq2Tree.trnn.TRNN attribute) (mwptoolkit.model.Seq2Tree.tsn.TSN attribute) (mwptoolkit.module.Attention.group_attention.GroupAttention attribute) (mwptoolkit.module.Attention.multi_head_attention.EPTMultiHeadAttention attribute) (mwptoolkit.module.Attention.multi_head_attention.EPTMultiHeadAttentionWeights attribute) (mwptoolkit.module.Attention.multi_head_attention.MultiHeadAttention attribute) (mwptoolkit.module.Attention.self_attention.SelfAttention attribute) (mwptoolkit.module.Attention.self_attention.SelfAttentionMask attribute) (mwptoolkit.module.Attention.seq_attention.Attention attribute) (mwptoolkit.module.Attention.seq_attention.MaskedRelevantScore attribute) (mwptoolkit.module.Attention.seq_attention.RelevantScore attribute) (mwptoolkit.module.Attention.seq_attention.SeqAttention attribute) (mwptoolkit.module.Attention.tree_attention.TreeAttention attribute) (mwptoolkit.module.Decoder.ept_decoder.AveragePooling attribute) (mwptoolkit.module.Decoder.ept_decoder.DecoderModel attribute) (mwptoolkit.module.Decoder.ept_decoder.ExpressionDecoderModel attribute) (mwptoolkit.module.Decoder.ept_decoder.ExpressionPointerTransformer attribute) (mwptoolkit.module.Decoder.ept_decoder.ExpressionTransformer attribute) (mwptoolkit.module.Decoder.ept_decoder.OpDecoderModel attribute) (mwptoolkit.module.Decoder.ept_decoder.Squeeze attribute) (mwptoolkit.module.Decoder.ept_decoder.VanillaOpTransformer attribute) (mwptoolkit.module.Decoder.rnn_decoder.AttentionalRNNDecoder attribute) (mwptoolkit.module.Decoder.rnn_decoder.BasicRNNDecoder attribute) (mwptoolkit.module.Decoder.rnn_decoder.SalignedDecoder attribute) (mwptoolkit.module.Decoder.transformer_decoder.TransformerDecoder attribute) (mwptoolkit.module.Decoder.tree_decoder.HMSDecoder attribute) (mwptoolkit.module.Decoder.tree_decoder.LSTMBasedTreeDecoder attribute) (mwptoolkit.module.Decoder.tree_decoder.PredictModel attribute) (mwptoolkit.module.Decoder.tree_decoder.RNNBasedTreeDecoder attribute) (mwptoolkit.module.Decoder.tree_decoder.SARTreeDecoder attribute) (mwptoolkit.module.Decoder.tree_decoder.TreeDecoder attribute) (mwptoolkit.module.Embedder.basic_embedder.BasicEmbedder attribute) (mwptoolkit.module.Embedder.bert_embedder.BertEmbedder attribute) (mwptoolkit.module.Embedder.position_embedder.DisPositionalEncoding attribute) (mwptoolkit.module.Embedder.position_embedder.EPTPositionalEncoding attribute) (mwptoolkit.module.Embedder.position_embedder.PositionalEncoding attribute) (mwptoolkit.module.Embedder.position_embedder.PositionEmbedder attribute) (mwptoolkit.module.Embedder.position_embedder.PositionEmbedder_x attribute) (mwptoolkit.module.Embedder.roberta_embedder.RobertaEmbedder attribute) (mwptoolkit.module.Encoder.graph_based_encoder.GraphBasedEncoder attribute) (mwptoolkit.module.Encoder.graph_based_encoder.GraphBasedMultiEncoder attribute) (mwptoolkit.module.Encoder.graph_based_encoder.GraphEncoder attribute) (mwptoolkit.module.Encoder.graph_based_encoder.NumEncoder attribute) (mwptoolkit.module.Encoder.rnn_encoder.BasicRNNEncoder attribute) (mwptoolkit.module.Encoder.rnn_encoder.GroupAttentionRNNEncoder attribute) (mwptoolkit.module.Encoder.rnn_encoder.HWCPEncoder attribute) (mwptoolkit.module.Encoder.rnn_encoder.SalignedEncoder attribute) (mwptoolkit.module.Encoder.rnn_encoder.SelfAttentionRNNEncoder attribute) (mwptoolkit.module.Encoder.transformer_encoder.BertEncoder attribute) (mwptoolkit.module.Encoder.transformer_encoder.GroupATTEncoder attribute) (mwptoolkit.module.Encoder.transformer_encoder.TransformerEncoder attribute) (mwptoolkit.module.Graph.gcn.GCN attribute) (mwptoolkit.module.Graph.graph_module.Graph_Module attribute) (mwptoolkit.module.Graph.graph_module.Num_Graph_Module attribute) (mwptoolkit.module.Graph.graph_module.Parse_Graph_Module attribute) (mwptoolkit.module.Layer.graph_layers.GraphConvolution attribute) (mwptoolkit.module.Layer.graph_layers.LayerNorm attribute) (mwptoolkit.module.Layer.graph_layers.MeanAggregator attribute) (mwptoolkit.module.Layer.graph_layers.PositionwiseFeedForward attribute) (mwptoolkit.module.Layer.layers.GenVar attribute) (mwptoolkit.module.Layer.layers.Transformer attribute) (mwptoolkit.module.Layer.layers.TreeAttnDecoderRNN attribute) (mwptoolkit.module.Layer.transformer_layer.EPTTransformerLayer attribute) (mwptoolkit.module.Layer.transformer_layer.GAEncoderLayer attribute) (mwptoolkit.module.Layer.transformer_layer.LayerNorm attribute) (mwptoolkit.module.Layer.transformer_layer.PositionwiseFeedForward attribute) (mwptoolkit.module.Layer.transformer_layer.SublayerConnection attribute) (mwptoolkit.module.Layer.transformer_layer.TransformerLayer attribute) (mwptoolkit.module.Layer.tree_layers.Dec_LSTM attribute) (mwptoolkit.module.Layer.tree_layers.DecomposeModel attribute) (mwptoolkit.module.Layer.tree_layers.DQN attribute) (mwptoolkit.module.Layer.tree_layers.GateNN attribute) (mwptoolkit.module.Layer.tree_layers.GenerateNode attribute) (mwptoolkit.module.Layer.tree_layers.Merge attribute) (mwptoolkit.module.Layer.tree_layers.NodeEmbeddingLayer attribute) (mwptoolkit.module.Layer.tree_layers.NodeGenerater attribute) (mwptoolkit.module.Layer.tree_layers.Prediction attribute) (mwptoolkit.module.Layer.tree_layers.RecursiveNN attribute) (mwptoolkit.module.Layer.tree_layers.Score attribute) (mwptoolkit.module.Layer.tree_layers.ScoreModel attribute) (mwptoolkit.module.Layer.tree_layers.SemanticAlignmentModule attribute) (mwptoolkit.module.Layer.tree_layers.SubTreeMerger attribute) (mwptoolkit.module.Layer.tree_layers.TreeAttention attribute) (mwptoolkit.module.Layer.tree_layers.TreeEmbeddingModel attribute)

trans_symbol_2_number() (in module mwptoolkit.utils.preprocess_tool.equation_operator) (in module mwptoolkit.utils.preprocess_tool.number_operator) Transformer (class in mwptoolkit.model.Seq2Seq.transformer) (class in mwptoolkit.module.Layer.layers) TransformerDecoder (class in mwptoolkit.module.Decoder.transformer_decoder) TransformerEncoder (class in mwptoolkit.module.Encoder.transformer_encoder) TransformerLayer (class in mwptoolkit.module.Layer.transformer_layer) traverse() (mwptoolkit.module.Layer.tree_layers.RecursiveNN method) Tree (class in mwptoolkit.utils.data_structure) tree2equ() (mwptoolkit.utils.data_structure.AbstractTree method) (mwptoolkit.utils.data_structure.BinaryTree method) tree2equation() (mwptoolkit.model.Seq2Tree.trnn.TRNN method) tree_decoder_forward() (mwptoolkit.model.Graph2Tree.multiencdec.MultiEncDec method) TreeAttention (class in mwptoolkit.module.Attention.tree_attention) (class in mwptoolkit.module.Layer.tree_layers) TreeAttnDecoderRNN (class in mwptoolkit.module.Layer.layers) TreeBeam (class in mwptoolkit.module.Strategy.beam_search) TreeDecoder (class in mwptoolkit.module.Decoder.tree_decoder) TreeEmbedding (class in mwptoolkit.module.Layer.tree_layers) TreeEmbeddingModel (class in mwptoolkit.module.Layer.tree_layers) TreeLSTM (class in mwptoolkit.model.Seq2Tree.treelstm) TreeLSTMTrainer (class in mwptoolkit.trainer.supervised_trainer) TreeNode (class in mwptoolkit.module.Layer.tree_layers) TRNN (class in mwptoolkit.model.Seq2Tree.trnn) TRNNTrainer (class in mwptoolkit.trainer.supervised_trainer) TSN (class in mwptoolkit.model.Seq2Tree.tsn) TSNTrainer (class in mwptoolkit.trainer.supervised_trainer)

U

UNK_TOKEN (mwptoolkit.utils.enum_type.SpecialTokens attribute)

V

Valid (mwptoolkit.utils.enum_type.DatasetType attribute) VanillaOpTransformer (class in mwptoolkit.module.Decoder.ept_decoder)

VAR_MAX (mwptoolkit.utils.enum_type.EPT attribute) VAR_PREFIX (mwptoolkit.utils.enum_type.EPT attribute)

W

weakly_supervised (mwptoolkit.utils.enum_type.SupervisingMode attribute)

word_level_forward() (mwptoolkit.module.Encoder.rnn_encoder.HWCPEncoder method) write_json_data() (in module mwptoolkit.utils.utils)

Z

zh (mwptoolkit.utils.enum_type.DatasetLanguage attribute)