Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/awslabs/gap-text2sql
/ functions
Functions
889 in github.com/awslabs/gap-text2sql
⨍
Functions
889
◇
Types & classes
177
Method
process_gen_token
(self, last_choice)
rat-sql-gap/seq2struct/models/nl2code/tree_traversal.py:376
Method
process_list_length_apply
(self, last_choice)
rat-sql-gap/seq2struct/models/nl2code/tree_traversal.py:334
Method
process_list_length_inquire
(self, last_choice)
rat-sql-gap/seq2struct/models/nl2code/tree_traversal.py:313
Method
process_node_finished
(self, last_choice)
rat-sql-gap/seq2struct/models/nl2code/tree_traversal.py:430
Method
process_pointer_apply
(self, last_choice)
rat-sql-gap/seq2struct/models/nl2code/tree_traversal.py:422
Method
process_pointer_inquire
(self, last_choice)
rat-sql-gap/seq2struct/models/nl2code/tree_traversal.py:399
Method
process_sum_apply
(self, last_choice)
rat-sql-gap/seq2struct/models/nl2code/tree_traversal.py:197
Method
process_sum_inquire
(self, last_choice)
rat-sql-gap/seq2struct/models/nl2code/tree_traversal.py:175
Method
rearranged_all_embs_gather_from_indices
(indices)
rat-sql-gap/seq2struct/models/spider/spider_enc_modules.py:245
Method
rearranged_all_embs_map_index
(desc_lengths_idx, seq_idx)
rat-sql-gap/seq2struct/models/spider/spider_enc_modules.py:242
Function
recall
(count, total)
rat-sql-gap/seq2struct/datasets/spider_lib/evaluation.py:92
Function
refine_schema_names
(schema)
rat-sql-gap/seq2struct/utils/api_utils.py:26
Function
register
(kind, name)
rat-sql-gap/seq2struct/utils/registry.py:10
Method
register_handler
(cls, func_type)
rat-sql-gap/seq2struct/models/nl2code/tree_traversal.py:36
Method
remove_product_type
(self, product_type_name)
rat-sql-gap/seq2struct/ast_util.py:148
Function
replace_masked_values
relogic/logickit/utils/utils.py:300
Method
replace_model_with_pretrained
(model, path, prefix)
rat-sql-gap/seq2struct/models/spider/spider_enc.py:1321
Method
resize_token_embeddings
(self, new_num_tokens: int)
relogic/pretrainkit/models/semparse/modeling_bart_copy.py:975
Method
resize_token_embeddings
(self, new_num_tokens: int)
relogic/pretrainkit/models/relationalsemparse/relational_semparse.py:30
Method
restore_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
Method
root_type
(self)
rat-sql-gap/seq2struct/ast_util.py:131
Method
rule_choice
(self, node_type, rule_logits)
rat-sql-gap/seq2struct/models/nl2code/train_tree_traversal.py:73
Method
rule_choice
(self, node_type, rule_logits)
rat-sql-gap/seq2struct/models/nl2code/infer_tree_traversal.py:76
Method
save
(self, out_path)
rat-sql-gap/seq2struct/utils/vocab.py:81
Method
save
(self, path)
rat-sql-gap/seq2struct/utils/vocab.py:108
Method
save
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
Method
save
(self)
rat-sql-gap/seq2struct/models/enc_dec.py:51
Method
save
(self)
rat-sql-gap/seq2struct/models/spider/spider_enc.py:249
Method
save
(self)
rat-sql-gap/seq2struct/models/spider/spider_enc.py:735
Method
save
(self)
rat-sql-gap/seq2struct/models/spider/spider_enc.py:1264
Method
save
(self)
rat-sql-gap/seq2struct/models/nl2code/decoder.py:140
Method
save
(self)
rat-sql-gap/seq2struct/models/nl2code/encoder.py:60
Method
save_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
Method
save_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
Method
save_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
Method
save_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
Method
save_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
Method
schema
(self)
rat-sql-gap/seq2struct/datasets/spider_lib/process_sql.py:57
Method
schema
(self)
rat-sql-gap/seq2struct/datasets/spider_lib/preprocess/schema.py:14
Method
select_subseq
(self, orig_batch_indices)
rat-sql-gap/seq2struct/utils/batched_sequence.py:111
Method
set_input_embeddings
(self, value)
relogic/pretrainkit/models/semparse/modeling_bart_copy.py:945
Method
set_input_embeddings
(self, value)
relogic/pretrainkit/models/relationalsemparse/modeling_relational_bart.py:982
Method
set_relation
(name)
relogic/pretrainkit/models/relationalsemparse/relational_transformer.py:639
Method
set_relation
(name)
rat-sql-gap/seq2struct/models/spider/spider_enc_modules.py:585
Function
sigmoid
Compute sigmoid values
relogic/logickit/utils/utils.py:27
Function
softmax
Compute softmax values for each sets of scores in x.
relogic/logickit/utils/utils.py:22
Function
sparse_attention
Compute 'Scaled Dot Product Attention
relogic/pretrainkit/models/relationalsemparse/relational_transformer.py:120
Function
sparse_attention
Compute 'Scaled Dot Product Attention
rat-sql-gap/seq2struct/models/transformer.py:110
Method
to
Transfer the pretrained embeddings to the given device.
rat-sql-gap/seq2struct/resources/pretrained_embeddings.py:39
Method
to
(self, device)
rat-sql-gap/seq2struct/resources/pretrained_embeddings.py:113
Function
to_dict_with_set_values
(d)
rat-sql-gap/seq2struct/utils/serialization.py:5
Function
to_dict_with_sorted_values
(d, key=None)
rat-sql-gap/seq2struct/utils/serialization.py:1
Method
to_str
(self)
rat-sql-gap/seq2struct/models/nl2code/tree_traversal.py:53
Method
token_choice
(self, output, gen_logodds)
rat-sql-gap/seq2struct/models/nl2code/train_tree_traversal.py:88
Method
token_choice
(self, output, gen_logodds)
rat-sql-gap/seq2struct/models/nl2code/infer_tree_traversal.py:79
Method
tokenize
Given a string, return a list of tokens suitable for lookup.
rat-sql-gap/seq2struct/resources/pretrained_embeddings.py:18
Method
tokenize
(self, text)
rat-sql-gap/seq2struct/resources/pretrained_embeddings.py:98
Function
torch_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
Function
torch_distributed_zero_first
Decorator to make all processes in distributed training wait for each local_master to do something.
relogic/pretrainkit/trainer.py:75
Method
train_batch_size
(self)
relogic/pretrainkit/training_args.py:146
Method
traverse_tree
(node)
rat-sql-gap/seq2struct/models/nl2code/decoder.py:528
Function
truncate_seq_pair
Truncates a sequence pair in place to the maximum length.
relogic/logickit/utils/utils.py:49
Function
tuplify
(x)
rat-sql-gap/seq2struct/utils/serialization.py:18
Method
types
(self)
rat-sql-gap/seq2struct/ast_util.py:126
Method
untokenize
Undo tokenize.
rat-sql-gap/seq2struct/resources/pretrained_embeddings.py:23
Method
untokenize
(self, tokens)
rat-sql-gap/seq2struct/resources/pretrained_embeddings.py:75
Method
untokenize
(self, tokens)
rat-sql-gap/seq2struct/resources/pretrained_embeddings.py:101
Method
update_lr
(self, current_step)
rat-sql-gap/seq2struct/optimizers.py:23
Method
update_lr
(self, current_step)
rat-sql-gap/seq2struct/optimizers.py:45
Method
update_lr
(self, current_step)
rat-sql-gap/seq2struct/optimizers.py:67
Method
update_lr
(self, current_step)
rat-sql-gap/seq2struct/optimizers.py:110
Method
update_using_last_choice
(self, last_choice, extra_choice_info, attention_offset)
rat-sql-gap/seq2struct/models/nl2code/train_tree_traversal.py:95
Method
update_using_last_choice
(self, last_choice, extra_choice_info, attention_offset)
rat-sql-gap/seq2struct/models/nl2code/infer_tree_traversal.py:99
Function
val_has_op
(val_unit)
rat-sql-gap/seq2struct/datasets/spider_lib/evaluation.py:78
Method
validate_item
Checks whether item can be successfully preprocessed. Returns a boolean and an arbitrary object.
rat-sql-gap/seq2struct/models/abstract_preproc.py:18
Method
validate_item
(self, item, section)
rat-sql-gap/seq2struct/models/spider/spider_enc.py:170
Method
validate_item
(self, item, section)
rat-sql-gap/seq2struct/models/spider/spider_enc.py:713
Method
validate_item
(self, item, section)
rat-sql-gap/seq2struct/models/spider/spider_enc.py:1243
Method
validate_item
(self, item, section)
rat-sql-gap/seq2struct/models/nl2code/decoder.py:118
Method
validate_item
(self, item, section)
rat-sql-gap/seq2struct/models/nl2code/encoder.py:45
Method
visitConstructor
(self, cons, _name)
rat-sql-gap/seq2struct/ast_util.py:43
Method
visitField
(self, field, name)
rat-sql-gap/seq2struct/ast_util.py:52
Method
visitModule
(self, mod)
rat-sql-gap/seq2struct/ast_util.py:28
Method
visitProduct
(self, prod, name)
rat-sql-gap/seq2struct/ast_util.py:59
Method
visitSum
(self, sum_, name)
rat-sql-gap/seq2struct/ast_util.py:37
Method
visitType
(self, type_)
rat-sql-gap/seq2struct/ast_util.py:33
Function
weighted_sum
Args: matrix (): attention ():
relogic/logickit/utils/utils.py:277
Method
with_new_ps
(self, ps)
rat-sql-gap/seq2struct/utils/batched_sequence.py:83
Function
write_pickle
(o, path)
relogic/logickit/base/utils.py:31
← previous
801–889 of 889, ranked by callers