MCPcopy Create free account

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

Functions889 in github.com/awslabs/gap-text2sql

↓ 284 callersMethodappend
(self, record)
rat-sql-gap/seq2struct/utils/indexed_file.py:22
↓ 106 callersMethodpop
(self)
rat-sql-gap/seq2struct/models/nl2code/tree_traversal.py:167
↓ 52 callersMethodto
(self, device)
rat-sql-gap/seq2struct/resources/pretrained_embeddings.py:87
↓ 49 callersMethodindex
(self, value)
rat-sql-gap/seq2struct/utils/vocab.py:62
↓ 40 callersMethodadd
(self, item, inferred_code, orig_question=None)
rat-sql-gap/seq2struct/datasets/spider.py:229
↓ 30 callersMethodclone
(self)
rat-sql-gap/seq2struct/models/nl2code/tree_traversal.py:103
↓ 29 callersMethodload
(self, in_path)
rat-sql-gap/seq2struct/utils/vocab.py:78
↓ 23 callersMethodsave
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 callersMethodstep
(self, last_choice, extra_choice_info=None, attention_offset=None)
rat-sql-gap/seq2struct/models/nl2code/tree_traversal.py:116
↓ 21 callersMethodtokenize
(self, text)
rat-sql-gap/seq2struct/resources/pretrained_embeddings.py:58
↓ 20 callersFunctionpad_and_tensorize_sequence
(sequences, padding_value)
relogic/pretrainkit/datasets/utils.py:3
↓ 18 callersMethodlog
(self, msg)
rat-sql-gap/seq2struct/commands/train.py:62
↓ 14 callersFunctionfilter_nones
(d)
rat-sql-gap/seq2struct/grammars/spider.py:18
↓ 13 callersMethodexpand
(self, k)
rat-sql-gap/seq2struct/utils/batched_sequence.py:137
↓ 13 callersMethodis_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 callersMethod__init__
(self, config: BartConfig)
relogic/pretrainkit/models/semparse/modeling_bart_copy.py:829
↓ 10 callersFunctionget_scores
(count, pred_total, label_total)
rat-sql-gap/seq2struct/datasets/spider_lib/evaluation.py:104
↓ 9 callersFunctionLayerNorm
(normalized_shape, eps=1e-5, elementwise_affine=True)
relogic/pretrainkit/models/semparse/modeling_bart_copy.py:798
↓ 9 callersFunctionLayerNorm
(normalized_shape, eps=1e-5, elementwise_affine=True)
relogic/pretrainkit/models/relationalsemparse/modeling_relational_bart.py:866
↓ 9 callersMethodsave_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 callersMethod__init__
(self, config: BartConfig, embed_tokens: nn.Embedding)
relogic/pretrainkit/models/relationalsemparse/modeling_relational_bart.py:516
↓ 8 callersMethodlookup
(self, token)
rat-sql-gap/seq2struct/resources/pretrained_embeddings.py:78
↓ 7 callersMethod__init__
(self, layer, layer_size, N, tie_layers=False)
relogic/pretrainkit/models/relationalsemparse/relational_transformer.py:276
↓ 7 callersMethodapply
(self, fn)
rat-sql-gap/seq2struct/utils/batched_sequence.py:79
↓ 7 callersMethodclose
(self)
rat-sql-gap/seq2struct/utils/indexed_file.py:27
↓ 7 callersMethoddevice
(self)
relogic/pretrainkit/training_args.py:198
↓ 7 callersMethodfrom_gather
(cls, lengths, map_index, gather_from_indices)
rat-sql-gap/seq2struct/utils/batched_sequence.py:181
↓ 7 callersMethodgenerate
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 callersMethodis_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 callersMethodpad_single_sentence_for_bert
(self, toks, cls=True)
rat-sql-gap/seq2struct/models/spider/spider_enc.py:994
↓ 7 callersMethodpad_single_sentence_for_bert
(self, toks, cls=True)
rat-sql-gap/seq2struct/models/spider/spider_enc.py:1536
↓ 7 callersMethodparse
(self, code, section)
rat-sql-gap/seq2struct/grammars/spider.py:122
↓ 7 callersMethodparse_cond
(self, cond, optional=False)
rat-sql-gap/seq2struct/grammars/spider.py:210
↓ 6 callersMethod__init__
(self, layer, layer_size, N, tie_layers=False)
rat-sql-gap/seq2struct/models/transformer.py:263
↓ 6 callersMethodcompute_loss
(self, enc_input, example, desc_enc, debug)
rat-sql-gap/seq2struct/models/nl2code/decoder.py:515
↓ 6 callersMethodfind_all_descendants_of_type
(self, tree, type, descend_pred=lambda field: True)
rat-sql-gap/seq2struct/ast_util.py:251
↓ 6 callersFunctionis_wandb_available
()
relogic/pretrainkit/multitask_trainer.py:60
↓ 6 callersMethodparse_col_unit
(self, col_unit)
rat-sql-gap/seq2struct/grammars/spider.py:174
↓ 6 callersMethodselect
(self, orig_batch_idx, seq_idx=None)
rat-sql-gap/seq2struct/utils/batched_sequence.py:105
↓ 5 callersMethod__init__
(self, pointer)
rat-sql-gap/seq2struct/models/attention.py:20
↓ 5 callersMethodannotate
(self, text, annotators=None, output_format=None, properties=None)
rat-sql-gap/seq2struct/resources/corenlp.py:25
↓ 5 callersFunctionbimap
(first, second)
rat-sql-gap/seq2struct/grammars/spider.py:14
↓ 5 callersFunctionclones
(module_fn, N)
relogic/pretrainkit/models/relationalsemparse/relational_transformer.py:102
↓ 5 callersFunctionclones
(module_fn, N)
rat-sql-gap/seq2struct/models/transformer.py:93
↓ 5 callersFunctioncount_agg
(units)
rat-sql-gap/seq2struct/datasets/spider_lib/evaluation.py:301
↓ 5 callersMethodgenerate_col_pred_data
(self, examples, padded_input_ids_tensor, use_type=False)
relogic/pretrainkit/datasets/semparse/tabart.py:133
↓ 5 callersMethodgenerate_mlm_data
(self, padded_input_ids_tensor, ignore_segment)
relogic/pretrainkit/datasets/semparse/tabart.py:122
↓ 5 callersMethodorig_lengths
(self)
rat-sql-gap/seq2struct/utils/batched_sequence.py:133
↓ 5 callersMethodresize_token_embeddings
(self, new_num_tokens: int)
relogic/pretrainkit/models/semparse/semparse.py:33
↓ 5 callersMethodstep
Performs a single optimization step. Arguments: closure (callable, optional): A closure that reevaluates the model
rat-sql-gap/seq2struct/optimizers.py:169
↓ 5 callersMethodunparse_col_unit
(self, col_unit)
rat-sql-gap/seq2struct/grammars/spider.py:433
↓ 4 callersMethod_tensor
(self, data, dtype=None)
rat-sql-gap/seq2struct/models/nl2code/decoder.py:698
↓ 4 callersMethod_update_state
( self, node_type, prev_state, prev_action_emb, pa
rat-sql-gap/seq2struct/models/nl2code/decoder.py:704
↓ 4 callersMethodadd
Add a new hypothesis to the list.
relogic/pretrainkit/models/semparse/semparse.py:1143
↓ 4 callersMethoddataset
(self, section)
rat-sql-gap/seq2struct/models/enc_dec.py:59
↓ 4 callersMethodevaluate
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 callersFunctionjoin
(iterable, delimiter)
rat-sql-gap/seq2struct/grammars/spider.py:22
↓ 4 callersFunctionlog
(*args)
relogic/logickit/base/utils.py:38
↓ 4 callersMethodnum_examples
Helper to get num of examples from a DataLoader, by accessing its Dataset.
relogic/pretrainkit/multitask_trainer.py:342
↓ 4 callersMethodnum_examples
Helper to get num of examples from a DataLoader, by accessing its Dataset.
relogic/pretrainkit/trainer.py:345
↓ 4 callersFunctionparse_col_unit
:returns next idx, (agg_op id, col_id)
rat-sql-gap/seq2struct/datasets/spider_lib/process_sql.py:191
↓ 4 callersMethodraw_index
(self, orig_batch_idx, seq_idx)
rat-sql-gap/seq2struct/utils/batched_sequence.py:98
↓ 4 callersFunctionrebuild_col_unit_col
(valid_col_units, col_unit, kmap)
rat-sql-gap/seq2struct/datasets/spider_lib/evaluation.py:711
↓ 4 callersFunctionrebuild_sql_val
(sql)
rat-sql-gap/seq2struct/datasets/spider_lib/evaluation.py:686
↓ 4 callersMethodsave_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 callersFunctionset_seed
(seed: int)
relogic/pretrainkit/trainer.py:66
↓ 4 callersMethodtrain
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 callersMethodunparse_val_unit
(self, val_unit)
rat-sql-gap/seq2struct/grammars/spider.py:451
↓ 3 callersMethod__init__
(self, pieces)
rat-sql-gap/seq2struct/models/spider/spider_enc.py:555
↓ 3 callersMethod__init__
(self, input_size, output_size, dropout, summarize, use_native=False)
rat-sql-gap/seq2struct/models/spider/spider_enc_modules.py:173
↓ 3 callersMethod_add_constructor
(self, sum_type_name, constructor)
rat-sql-gap/seq2struct/ast_util.py:163
↓ 3 callersMethod_bert_encode
(self, toks)
rat-sql-gap/seq2struct/models/spider/spider_enc.py:970
↓ 3 callersMethod_bert_encode
(self, toks)
rat-sql-gap/seq2struct/models/spider/spider_enc.py:1512
↓ 3 callersMethod_build_modules
(self, module_types)
rat-sql-gap/seq2struct/models/spider/spider_enc.py:327
↓ 3 callersMethod_index
(self, vocab, word)
rat-sql-gap/seq2struct/models/nl2code/decoder.py:701
↓ 3 callersFunction_make_linear_from_emb
(emb)
relogic/pretrainkit/models/semparse/modeling_bart_copy.py:170
↓ 3 callersFunction_make_packed_sequence
(data, batch_sizes)
rat-sql-gap/seq2struct/utils/batched_sequence.py:59
↓ 3 callersMethod_shape
(self, tensor, dim_0, bsz)
relogic/pretrainkit/models/semparse/modeling_bart_copy.py:606
↓ 3 callersMethod_shape
(self, tensor, dim_0, bsz)
relogic/pretrainkit/models/relationalsemparse/modeling_relational_bart.py:674
↓ 3 callersMethodadd_word
(self, word, count=1)
rat-sql-gap/seq2struct/utils/vocab.py:92
↓ 3 callersMethodapply_rule
( self, node_type, prev_state, prev_action_emb, pa
rat-sql-gap/seq2struct/models/nl2code/decoder.py:735
↓ 3 callersMethodbegin_inference
(self, desc_enc, example)
rat-sql-gap/seq2struct/models/nl2code/decoder.py:681
↓ 3 callersFunctionclamp
(value, abs_max)
relogic/pretrainkit/models/relationalsemparse/relational_transformer.py:12
↓ 3 callersFunctionclamp
(value, abs_max)
rat-sql-gap/seq2struct/models/spider/spider_enc_modules.py:14
↓ 3 callersMethodcolumn_prediction
(self, input_ids, attention_mask, column_spans)
relogic/pretrainkit/models/semparse/logical_tabart.py:54
↓ 3 callersFunctioncompute_cell_value_linking
(tokens, schema, db_dir)
rat-sql-gap/seq2struct/models/spider/spider_match_utils.py:80
↓ 3 callersFunctioncompute_schema_linking
(question, column, table)
rat-sql-gap/seq2struct/models/spider/spider_match_utils.py:13
↓ 3 callersFunctioneval_nested
(pred, label)
rat-sql-gap/seq2struct/datasets/spider_lib/evaluation.py:223
↓ 3 callersMethodevaluate_one
(self, db_name, gold, predicted)
rat-sql-gap/seq2struct/datasets/spider_lib/evaluation.py:456
↓ 3 callersMethodfinalize
(self)
rat-sql-gap/seq2struct/datasets/spider.py:248
↓ 3 callersMethodfinish
(self, *args, **kwargs)
rat-sql-gap/seq2struct/utils/vocab.py:95
↓ 3 callersMethodforward_unbatched
(self, desc)
rat-sql-gap/seq2struct/models/spider/spider_enc.py:368
↓ 3 callersFunctionget_field_presence_info
(ast_wrapper, node, field_infos)
rat-sql-gap/seq2struct/models/nl2code/decoder.py:61
↓ 3 callersFunctionget_sql
(schema, query)
rat-sql-gap/seq2struct/datasets/spider_lib/process_sql.py:550
↓ 3 callersFunctioninvert_mask
(attention_mask)
relogic/pretrainkit/models/semparse/modeling_bart_copy.py:115
↓ 3 callersFunctioninvert_mask
(attention_mask)
relogic/pretrainkit/models/relationalsemparse/modeling_relational_bart.py:114
↓ 3 callersMethodis_local_master
(self)
relogic/pretrainkit/trainer.py:605
↓ 3 callersFunctionpad_and_tensorize_sequence
(sequences, padding_value)
relogic/pretrainkit/datasets/semparse/column_inferring.py:13
↓ 3 callersFunctionpad_and_tensorize_sequence
(sequences, padding_value)
relogic/pretrainkit/datasets/semparse/text2sql.py:20
↓ 3 callersFunctionparse_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