MCPcopy Create free account

hub / github.com/awslabs/gap-text2sql / functions

Functions889 in github.com/awslabs/gap-text2sql

Methodprocess_gen_token
(self, last_choice)
rat-sql-gap/seq2struct/models/nl2code/tree_traversal.py:376
Methodprocess_list_length_apply
(self, last_choice)
rat-sql-gap/seq2struct/models/nl2code/tree_traversal.py:334
Methodprocess_list_length_inquire
(self, last_choice)
rat-sql-gap/seq2struct/models/nl2code/tree_traversal.py:313
Methodprocess_node_finished
(self, last_choice)
rat-sql-gap/seq2struct/models/nl2code/tree_traversal.py:430
Methodprocess_pointer_apply
(self, last_choice)
rat-sql-gap/seq2struct/models/nl2code/tree_traversal.py:422
Methodprocess_pointer_inquire
(self, last_choice)
rat-sql-gap/seq2struct/models/nl2code/tree_traversal.py:399
Methodprocess_sum_apply
(self, last_choice)
rat-sql-gap/seq2struct/models/nl2code/tree_traversal.py:197
Methodprocess_sum_inquire
(self, last_choice)
rat-sql-gap/seq2struct/models/nl2code/tree_traversal.py:175
Methodrearranged_all_embs_gather_from_indices
(indices)
rat-sql-gap/seq2struct/models/spider/spider_enc_modules.py:245
Methodrearranged_all_embs_map_index
(desc_lengths_idx, seq_idx)
rat-sql-gap/seq2struct/models/spider/spider_enc_modules.py:242
Functionrecall
(count, total)
rat-sql-gap/seq2struct/datasets/spider_lib/evaluation.py:92
Functionrefine_schema_names
(schema)
rat-sql-gap/seq2struct/utils/api_utils.py:26
Functionregister
(kind, name)
rat-sql-gap/seq2struct/utils/registry.py:10
Methodregister_handler
(cls, func_type)
rat-sql-gap/seq2struct/models/nl2code/tree_traversal.py:36
Methodremove_product_type
(self, product_type_name)
rat-sql-gap/seq2struct/ast_util.py:148
Functionreplace_masked_values
relogic/logickit/utils/utils.py:300
Methodreplace_model_with_pretrained
(model, path, prefix)
rat-sql-gap/seq2struct/models/spider/spider_enc.py:1321
Methodresize_token_embeddings
(self, new_num_tokens: int)
relogic/pretrainkit/models/semparse/modeling_bart_copy.py:975
Methodresize_token_embeddings
(self, new_num_tokens: int)
relogic/pretrainkit/models/relationalsemparse/relational_semparse.py:30
Methodrestore_part
Restores part of the model from other directory. Useful to initialize part of the model with another pretrained model. Args:
rat-sql-gap/seq2struct/utils/saver.py:135
Methodroot_type
(self)
rat-sql-gap/seq2struct/ast_util.py:131
Methodrule_choice
(self, node_type, rule_logits)
rat-sql-gap/seq2struct/models/nl2code/train_tree_traversal.py:73
Methodrule_choice
(self, node_type, rule_logits)
rat-sql-gap/seq2struct/models/nl2code/infer_tree_traversal.py:76
Methodsave
(self, out_path)
rat-sql-gap/seq2struct/utils/vocab.py:81
Methodsave
(self, path)
rat-sql-gap/seq2struct/utils/vocab.py:108
Methodsave
Marks that all of the items have been preprocessed. Save state to disk. Used in preprocess.py, after reading all of the data.
rat-sql-gap/seq2struct/models/abstract_preproc.py:35
Methodsave
(self)
rat-sql-gap/seq2struct/models/enc_dec.py:51
Methodsave
(self)
rat-sql-gap/seq2struct/models/spider/spider_enc.py:249
Methodsave
(self)
rat-sql-gap/seq2struct/models/spider/spider_enc.py:735
Methodsave
(self)
rat-sql-gap/seq2struct/models/spider/spider_enc.py:1264
Methodsave
(self)
rat-sql-gap/seq2struct/models/nl2code/decoder.py:140
Methodsave
(self)
rat-sql-gap/seq2struct/models/nl2code/encoder.py:60
Methodsave_pretrained
Save a model and its configuration file to a directory, so that it can be re-loaded using the `:func:`~transformers.PreTrainedModel.from_pret
relogic/pretrainkit/models/multid/multid.py:62
Methodsave_pretrained
Save a model and its configuration file to a directory, so that it can be re-loaded using the `:func:`~transformers.PreTrainedModel.from_pret
relogic/pretrainkit/models/semparse/bart_parser.py:70
Methodsave_pretrained
Save a model and its configuration file to a directory, so that it can be re-loaded using the `:func:`~transformers.PreTrainedModel.from_pret
relogic/pretrainkit/models/semparse/logical_tabart.py:334
Methodsave_pretrained
Save a model and its configuration file to a directory, so that it can be re-loaded using the `:func:`~transformers.PreTrainedModel.from_pret
relogic/pretrainkit/models/semparse/tabart.py:112
Methodsave_pretrained
Save a model and its configuration file to a directory, so that it can be re-loaded using the `:func:`~transformers.PreTrainedModel.from_pret
relogic/pretrainkit/models/relationalsemparse/relational_bart_parser.py:72
Methodschema
(self)
rat-sql-gap/seq2struct/datasets/spider_lib/process_sql.py:57
Methodschema
(self)
rat-sql-gap/seq2struct/datasets/spider_lib/preprocess/schema.py:14
Methodselect_subseq
(self, orig_batch_indices)
rat-sql-gap/seq2struct/utils/batched_sequence.py:111
Methodset_input_embeddings
(self, value)
relogic/pretrainkit/models/semparse/modeling_bart_copy.py:945
Methodset_input_embeddings
(self, value)
relogic/pretrainkit/models/relationalsemparse/modeling_relational_bart.py:982
Methodset_relation
(name)
relogic/pretrainkit/models/relationalsemparse/relational_transformer.py:639
Methodset_relation
(name)
rat-sql-gap/seq2struct/models/spider/spider_enc_modules.py:585
Functionsigmoid
Compute sigmoid values
relogic/logickit/utils/utils.py:27
Functionsoftmax
Compute softmax values for each sets of scores in x.
relogic/logickit/utils/utils.py:22
Functionsparse_attention
Compute 'Scaled Dot Product Attention
relogic/pretrainkit/models/relationalsemparse/relational_transformer.py:120
Functionsparse_attention
Compute 'Scaled Dot Product Attention
rat-sql-gap/seq2struct/models/transformer.py:110
Methodto
Transfer the pretrained embeddings to the given device.
rat-sql-gap/seq2struct/resources/pretrained_embeddings.py:39
Methodto
(self, device)
rat-sql-gap/seq2struct/resources/pretrained_embeddings.py:113
Functionto_dict_with_set_values
(d)
rat-sql-gap/seq2struct/utils/serialization.py:5
Functionto_dict_with_sorted_values
(d, key=None)
rat-sql-gap/seq2struct/utils/serialization.py:1
Methodto_str
(self)
rat-sql-gap/seq2struct/models/nl2code/tree_traversal.py:53
Methodtoken_choice
(self, output, gen_logodds)
rat-sql-gap/seq2struct/models/nl2code/train_tree_traversal.py:88
Methodtoken_choice
(self, output, gen_logodds)
rat-sql-gap/seq2struct/models/nl2code/infer_tree_traversal.py:79
Methodtokenize
Given a string, return a list of tokens suitable for lookup.
rat-sql-gap/seq2struct/resources/pretrained_embeddings.py:18
Methodtokenize
(self, text)
rat-sql-gap/seq2struct/resources/pretrained_embeddings.py:98
Functiontorch_distributed_zero_first
Decorator to make all processes in distributed training wait for each local_master to do something.
relogic/pretrainkit/multitask_trainer.py:79
Functiontorch_distributed_zero_first
Decorator to make all processes in distributed training wait for each local_master to do something.
relogic/pretrainkit/trainer.py:75
Methodtrain_batch_size
(self)
relogic/pretrainkit/training_args.py:146
Methodtraverse_tree
(node)
rat-sql-gap/seq2struct/models/nl2code/decoder.py:528
Functiontruncate_seq_pair
Truncates a sequence pair in place to the maximum length.
relogic/logickit/utils/utils.py:49
Functiontuplify
(x)
rat-sql-gap/seq2struct/utils/serialization.py:18
Methodtypes
(self)
rat-sql-gap/seq2struct/ast_util.py:126
Methoduntokenize
Undo tokenize.
rat-sql-gap/seq2struct/resources/pretrained_embeddings.py:23
Methoduntokenize
(self, tokens)
rat-sql-gap/seq2struct/resources/pretrained_embeddings.py:75
Methoduntokenize
(self, tokens)
rat-sql-gap/seq2struct/resources/pretrained_embeddings.py:101
Methodupdate_lr
(self, current_step)
rat-sql-gap/seq2struct/optimizers.py:23
Methodupdate_lr
(self, current_step)
rat-sql-gap/seq2struct/optimizers.py:45
Methodupdate_lr
(self, current_step)
rat-sql-gap/seq2struct/optimizers.py:67
Methodupdate_lr
(self, current_step)
rat-sql-gap/seq2struct/optimizers.py:110
Methodupdate_using_last_choice
(self, last_choice, extra_choice_info, attention_offset)
rat-sql-gap/seq2struct/models/nl2code/train_tree_traversal.py:95
Methodupdate_using_last_choice
(self, last_choice, extra_choice_info, attention_offset)
rat-sql-gap/seq2struct/models/nl2code/infer_tree_traversal.py:99
Functionval_has_op
(val_unit)
rat-sql-gap/seq2struct/datasets/spider_lib/evaluation.py:78
Methodvalidate_item
Checks whether item can be successfully preprocessed. Returns a boolean and an arbitrary object.
rat-sql-gap/seq2struct/models/abstract_preproc.py:18
Methodvalidate_item
(self, item, section)
rat-sql-gap/seq2struct/models/spider/spider_enc.py:170
Methodvalidate_item
(self, item, section)
rat-sql-gap/seq2struct/models/spider/spider_enc.py:713
Methodvalidate_item
(self, item, section)
rat-sql-gap/seq2struct/models/spider/spider_enc.py:1243
Methodvalidate_item
(self, item, section)
rat-sql-gap/seq2struct/models/nl2code/decoder.py:118
Methodvalidate_item
(self, item, section)
rat-sql-gap/seq2struct/models/nl2code/encoder.py:45
MethodvisitConstructor
(self, cons, _name)
rat-sql-gap/seq2struct/ast_util.py:43
MethodvisitField
(self, field, name)
rat-sql-gap/seq2struct/ast_util.py:52
MethodvisitModule
(self, mod)
rat-sql-gap/seq2struct/ast_util.py:28
MethodvisitProduct
(self, prod, name)
rat-sql-gap/seq2struct/ast_util.py:59
MethodvisitSum
(self, sum_, name)
rat-sql-gap/seq2struct/ast_util.py:37
MethodvisitType
(self, type_)
rat-sql-gap/seq2struct/ast_util.py:33
Functionweighted_sum
Args: matrix (): attention ():
relogic/logickit/utils/utils.py:277
Methodwith_new_ps
(self, ps)
rat-sql-gap/seq2struct/utils/batched_sequence.py:83
Functionwrite_pickle
(o, path)
relogic/logickit/base/utils.py:31
← previous801–889 of 889, ranked by callers