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
↓ 284 callers
Method
append
(self, record)
rat-sql-gap/seq2struct/utils/indexed_file.py:22
↓ 106 callers
Method
pop
(self)
rat-sql-gap/seq2struct/models/nl2code/tree_traversal.py:167
↓ 52 callers
Method
to
(self, device)
rat-sql-gap/seq2struct/resources/pretrained_embeddings.py:87
↓ 49 callers
Method
index
(self, value)
rat-sql-gap/seq2struct/utils/vocab.py:62
↓ 40 callers
Method
add
(self, item, inferred_code, orig_question=None)
rat-sql-gap/seq2struct/datasets/spider.py:229
↓ 30 callers
Method
clone
(self)
rat-sql-gap/seq2struct/models/nl2code/tree_traversal.py:103
↓ 29 callers
Method
load
(self, in_path)
rat-sql-gap/seq2struct/utils/vocab.py:78
↓ 23 callers
Method
save
Saves model and optimizer to given directory. Args: model_dir: Model directory to save. step: Current training step.
rat-sql-gap/seq2struct/utils/saver.py:125
↓ 21 callers
Method
step
(self, last_choice, extra_choice_info=None, attention_offset=None)
rat-sql-gap/seq2struct/models/nl2code/tree_traversal.py:116
↓ 21 callers
Method
tokenize
(self, text)
rat-sql-gap/seq2struct/resources/pretrained_embeddings.py:58
↓ 20 callers
Function
pad_and_tensorize_sequence
(sequences, padding_value)
relogic/pretrainkit/datasets/utils.py:3
↓ 18 callers
Method
log
(self, msg)
rat-sql-gap/seq2struct/commands/train.py:62
↓ 14 callers
Function
filter_nones
(d)
rat-sql-gap/seq2struct/grammars/spider.py:18
↓ 13 callers
Method
expand
(self, k)
rat-sql-gap/seq2struct/utils/batched_sequence.py:137
↓ 13 callers
Method
is_world_master
This will be True only in one process, even in distributed mode, even when training on multiple machines.
relogic/pretrainkit/trainer.py:611
↓ 10 callers
Method
__init__
(self, config: BartConfig)
relogic/pretrainkit/models/semparse/modeling_bart_copy.py:829
↓ 10 callers
Function
get_scores
(count, pred_total, label_total)
rat-sql-gap/seq2struct/datasets/spider_lib/evaluation.py:104
↓ 9 callers
Function
LayerNorm
(normalized_shape, eps=1e-5, elementwise_affine=True)
relogic/pretrainkit/models/semparse/modeling_bart_copy.py:798
↓ 9 callers
Function
LayerNorm
(normalized_shape, eps=1e-5, elementwise_affine=True)
relogic/pretrainkit/models/relationalsemparse/modeling_relational_bart.py:866
↓ 9 callers
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/sql_to_text.py:62
↓ 8 callers
Method
__init__
(self, config: BartConfig, embed_tokens: nn.Embedding)
relogic/pretrainkit/models/relationalsemparse/modeling_relational_bart.py:516
↓ 8 callers
Method
lookup
(self, token)
rat-sql-gap/seq2struct/resources/pretrained_embeddings.py:78
↓ 7 callers
Method
__init__
(self, layer, layer_size, N, tie_layers=False)
relogic/pretrainkit/models/relationalsemparse/relational_transformer.py:276
↓ 7 callers
Method
apply
(self, fn)
rat-sql-gap/seq2struct/utils/batched_sequence.py:79
↓ 7 callers
Method
close
(self)
rat-sql-gap/seq2struct/utils/indexed_file.py:27
↓ 7 callers
Method
device
(self)
relogic/pretrainkit/training_args.py:198
↓ 7 callers
Method
from_gather
(cls, lengths, map_index, gather_from_indices)
rat-sql-gap/seq2struct/utils/batched_sequence.py:181
↓ 7 callers
Method
generate
r""" Generates sequences for models with a LM head. The method currently supports greedy decoding, beam-search decoding, sampling with temperature, sa
relogic/pretrainkit/models/semparse/semparse.py:209
↓ 7 callers
Method
is_world_master
This will be True only in one process, even in distributed mode, even when training on multiple machines.
relogic/pretrainkit/multitask_trainer.py:632
↓ 7 callers
Method
pad_single_sentence_for_bert
(self, toks, cls=True)
rat-sql-gap/seq2struct/models/spider/spider_enc.py:994
↓ 7 callers
Method
pad_single_sentence_for_bert
(self, toks, cls=True)
rat-sql-gap/seq2struct/models/spider/spider_enc.py:1536
↓ 7 callers
Method
parse
(self, code, section)
rat-sql-gap/seq2struct/grammars/spider.py:122
↓ 7 callers
Method
parse_cond
(self, cond, optional=False)
rat-sql-gap/seq2struct/grammars/spider.py:210
↓ 6 callers
Method
__init__
(self, layer, layer_size, N, tie_layers=False)
rat-sql-gap/seq2struct/models/transformer.py:263
↓ 6 callers
Method
compute_loss
(self, enc_input, example, desc_enc, debug)
rat-sql-gap/seq2struct/models/nl2code/decoder.py:515
↓ 6 callers
Method
find_all_descendants_of_type
(self, tree, type, descend_pred=lambda field: True)
rat-sql-gap/seq2struct/ast_util.py:251
↓ 6 callers
Function
is_wandb_available
()
relogic/pretrainkit/multitask_trainer.py:60
↓ 6 callers
Method
parse_col_unit
(self, col_unit)
rat-sql-gap/seq2struct/grammars/spider.py:174
↓ 6 callers
Method
select
(self, orig_batch_idx, seq_idx=None)
rat-sql-gap/seq2struct/utils/batched_sequence.py:105
↓ 5 callers
Method
__init__
(self, pointer)
rat-sql-gap/seq2struct/models/attention.py:20
↓ 5 callers
Method
annotate
(self, text, annotators=None, output_format=None, properties=None)
rat-sql-gap/seq2struct/resources/corenlp.py:25
↓ 5 callers
Function
bimap
(first, second)
rat-sql-gap/seq2struct/grammars/spider.py:14
↓ 5 callers
Function
clones
(module_fn, N)
relogic/pretrainkit/models/relationalsemparse/relational_transformer.py:102
↓ 5 callers
Function
clones
(module_fn, N)
rat-sql-gap/seq2struct/models/transformer.py:93
↓ 5 callers
Function
count_agg
(units)
rat-sql-gap/seq2struct/datasets/spider_lib/evaluation.py:301
↓ 5 callers
Method
generate_col_pred_data
(self, examples, padded_input_ids_tensor, use_type=False)
relogic/pretrainkit/datasets/semparse/tabart.py:133
↓ 5 callers
Method
generate_mlm_data
(self, padded_input_ids_tensor, ignore_segment)
relogic/pretrainkit/datasets/semparse/tabart.py:122
↓ 5 callers
Method
orig_lengths
(self)
rat-sql-gap/seq2struct/utils/batched_sequence.py:133
↓ 5 callers
Method
resize_token_embeddings
(self, new_num_tokens: int)
relogic/pretrainkit/models/semparse/semparse.py:33
↓ 5 callers
Method
step
Performs a single optimization step. Arguments: closure (callable, optional): A closure that reevaluates the model
rat-sql-gap/seq2struct/optimizers.py:169
↓ 5 callers
Method
unparse_col_unit
(self, col_unit)
rat-sql-gap/seq2struct/grammars/spider.py:433
↓ 4 callers
Method
_tensor
(self, data, dtype=None)
rat-sql-gap/seq2struct/models/nl2code/decoder.py:698
↓ 4 callers
Method
_update_state
( self, node_type, prev_state, prev_action_emb, pa
rat-sql-gap/seq2struct/models/nl2code/decoder.py:704
↓ 4 callers
Method
add
Add a new hypothesis to the list.
relogic/pretrainkit/models/semparse/semparse.py:1143
↓ 4 callers
Method
dataset
(self, section)
rat-sql-gap/seq2struct/models/enc_dec.py:59
↓ 4 callers
Method
evaluate
Run evaluation and return metrics. The calling script will be responsible for providing a method to compute metrics, as they are
relogic/pretrainkit/trainer.py:695
↓ 4 callers
Function
join
(iterable, delimiter)
rat-sql-gap/seq2struct/grammars/spider.py:22
↓ 4 callers
Function
log
(*args)
relogic/logickit/base/utils.py:38
↓ 4 callers
Method
num_examples
Helper to get num of examples from a DataLoader, by accessing its Dataset.
relogic/pretrainkit/multitask_trainer.py:342
↓ 4 callers
Method
num_examples
Helper to get num of examples from a DataLoader, by accessing its Dataset.
relogic/pretrainkit/trainer.py:345
↓ 4 callers
Function
parse_col_unit
:returns next idx, (agg_op id, col_id)
rat-sql-gap/seq2struct/datasets/spider_lib/process_sql.py:191
↓ 4 callers
Method
raw_index
(self, orig_batch_idx, seq_idx)
rat-sql-gap/seq2struct/utils/batched_sequence.py:98
↓ 4 callers
Function
rebuild_col_unit_col
(valid_col_units, col_unit, kmap)
rat-sql-gap/seq2struct/datasets/spider_lib/evaluation.py:711
↓ 4 callers
Function
rebuild_sql_val
(sql)
rat-sql-gap/seq2struct/datasets/spider_lib/evaluation.py:686
↓ 4 callers
Method
save_model
Saving best-practices: if you use default names for the model, you can reload it using from_pretrained(). Will only save fro
relogic/pretrainkit/trainer.py:621
↓ 4 callers
Function
set_seed
(seed: int)
relogic/pretrainkit/trainer.py:66
↓ 4 callers
Method
train
Main training entry point. Args: model_path: (Optional) Local path to model if model to train has been i
relogic/pretrainkit/trainer.py:351
↓ 4 callers
Method
unparse_val_unit
(self, val_unit)
rat-sql-gap/seq2struct/grammars/spider.py:451
↓ 3 callers
Method
__init__
(self, pieces)
rat-sql-gap/seq2struct/models/spider/spider_enc.py:555
↓ 3 callers
Method
__init__
(self, input_size, output_size, dropout, summarize, use_native=False)
rat-sql-gap/seq2struct/models/spider/spider_enc_modules.py:173
↓ 3 callers
Method
_add_constructor
(self, sum_type_name, constructor)
rat-sql-gap/seq2struct/ast_util.py:163
↓ 3 callers
Method
_bert_encode
(self, toks)
rat-sql-gap/seq2struct/models/spider/spider_enc.py:970
↓ 3 callers
Method
_bert_encode
(self, toks)
rat-sql-gap/seq2struct/models/spider/spider_enc.py:1512
↓ 3 callers
Method
_build_modules
(self, module_types)
rat-sql-gap/seq2struct/models/spider/spider_enc.py:327
↓ 3 callers
Method
_index
(self, vocab, word)
rat-sql-gap/seq2struct/models/nl2code/decoder.py:701
↓ 3 callers
Function
_make_linear_from_emb
(emb)
relogic/pretrainkit/models/semparse/modeling_bart_copy.py:170
↓ 3 callers
Function
_make_packed_sequence
(data, batch_sizes)
rat-sql-gap/seq2struct/utils/batched_sequence.py:59
↓ 3 callers
Method
_shape
(self, tensor, dim_0, bsz)
relogic/pretrainkit/models/semparse/modeling_bart_copy.py:606
↓ 3 callers
Method
_shape
(self, tensor, dim_0, bsz)
relogic/pretrainkit/models/relationalsemparse/modeling_relational_bart.py:674
↓ 3 callers
Method
add_word
(self, word, count=1)
rat-sql-gap/seq2struct/utils/vocab.py:92
↓ 3 callers
Method
apply_rule
( self, node_type, prev_state, prev_action_emb, pa
rat-sql-gap/seq2struct/models/nl2code/decoder.py:735
↓ 3 callers
Method
begin_inference
(self, desc_enc, example)
rat-sql-gap/seq2struct/models/nl2code/decoder.py:681
↓ 3 callers
Function
clamp
(value, abs_max)
relogic/pretrainkit/models/relationalsemparse/relational_transformer.py:12
↓ 3 callers
Function
clamp
(value, abs_max)
rat-sql-gap/seq2struct/models/spider/spider_enc_modules.py:14
↓ 3 callers
Method
column_prediction
(self, input_ids, attention_mask, column_spans)
relogic/pretrainkit/models/semparse/logical_tabart.py:54
↓ 3 callers
Function
compute_cell_value_linking
(tokens, schema, db_dir)
rat-sql-gap/seq2struct/models/spider/spider_match_utils.py:80
↓ 3 callers
Function
compute_schema_linking
(question, column, table)
rat-sql-gap/seq2struct/models/spider/spider_match_utils.py:13
↓ 3 callers
Function
eval_nested
(pred, label)
rat-sql-gap/seq2struct/datasets/spider_lib/evaluation.py:223
↓ 3 callers
Method
evaluate_one
(self, db_name, gold, predicted)
rat-sql-gap/seq2struct/datasets/spider_lib/evaluation.py:456
↓ 3 callers
Method
finalize
(self)
rat-sql-gap/seq2struct/datasets/spider.py:248
↓ 3 callers
Method
finish
(self, *args, **kwargs)
rat-sql-gap/seq2struct/utils/vocab.py:95
↓ 3 callers
Method
forward_unbatched
(self, desc)
rat-sql-gap/seq2struct/models/spider/spider_enc.py:368
↓ 3 callers
Function
get_field_presence_info
(ast_wrapper, node, field_infos)
rat-sql-gap/seq2struct/models/nl2code/decoder.py:61
↓ 3 callers
Function
get_sql
(schema, query)
rat-sql-gap/seq2struct/datasets/spider_lib/process_sql.py:550
↓ 3 callers
Function
invert_mask
(attention_mask)
relogic/pretrainkit/models/semparse/modeling_bart_copy.py:115
↓ 3 callers
Function
invert_mask
(attention_mask)
relogic/pretrainkit/models/relationalsemparse/modeling_relational_bart.py:114
↓ 3 callers
Method
is_local_master
(self)
relogic/pretrainkit/trainer.py:605
↓ 3 callers
Function
pad_and_tensorize_sequence
(sequences, padding_value)
relogic/pretrainkit/datasets/semparse/column_inferring.py:13
↓ 3 callers
Function
pad_and_tensorize_sequence
(sequences, padding_value)
relogic/pretrainkit/datasets/semparse/text2sql.py:20
↓ 3 callers
Function
parse_condition
(toks, start_idx, tables_with_alias, schema, default_tables=None)
rat-sql-gap/seq2struct/datasets/spider_lib/process_sql.py:304
next →
1–100 of 889, ranked by callers