MCPcopy Create free account

hub / github.com/Snowflake-Labs/ReFoRCE / functions

Functions554 in github.com/Snowflake-Labs/ReFoRCE

Functionbuild_foreign_key_map
(entry)
spider2-lite/baselines/dailsql/utils/datasets/spider.py:14
Methodcheck_length
(prompt_components, new)
spider2-lite/baselines/dailsql/prompt/PromptReprTemplate.py:64
Methodclear_items
(self)
spider2-lite/baselines/dailsql/utils/linking_process.py:152
Methodclear_items
Clear the preprocessed items
spider2-lite/baselines/dailsql/utils/linking_utils/abstract_preproc.py:29
Functioncollect_result
(result)
spider2-lite/baselines/dailsql/generate_question.py:39
Functioncompare_multi_pandas_table
(pred, multi_gold, multi_condition_cols=[], multi_ignore_order=False)
spider2-lite/evaluation_suite/evaluate_utils.py:60
Methodcompute_loss
( self, batch_table_name_cls_logits, batch_table_labels, batch_column_info_cls
spider2-lite/baselines/codes/utils/classifier_loss.py:51
Methodcontains
(self, token)
spider2-lite/baselines/dailsql/utils/pretrained_embeddings.py:34
Methodcontains
(self, token)
spider2-lite/baselines/dailsql/utils/pretrained_embeddings.py:102
Methodcreate_custom_forward
(module)
spider2-lite/baselines/codes/modeling_gpt_bigcode.py:753
Methodcustom_forward
(*inputs)
spider2-lite/baselines/codes/modeling_gpt_bigcode.py:754
Methoddataset
(self, section)
spider2-lite/baselines/dailsql/utils/linking_process.py:231
Methoddataset
Returns a torch.data.utils.Dataset instance.
spider2-lite/baselines/dailsql/utils/linking_utils/abstract_preproc.py:45
Functiondeduplicate_by_value
(input_dict)
methods/SL/gen_sl/utils.py:44
Functiondigit_entropy_ratio
(s: str)
methods/ReFoRCE/utils.py:289
Methoddomain_mask
(self, candidates: list, db_id)
spider2-lite/baselines/dailsql/prompt/ExampleSelectorTemplate.py:19
Functioneasy_prompt_maker
(test_sample_text, db_names, schema_links, spider_schema, spider_foreign)
spider2-lite/baselines/dinsql/DIN-SQL.py:577
Functionevaluate_bird
(gold_result_dir, exec_result_path, example_id, task=None)
methods/ReFoRCE/eval.py:120
Functionevaluate_instance
Function to evaluate each instance, to be executed in parallel.
spider2-snow/evaluation_suite/evaluate_yx.py:20
Functionevaluate_single_exec_result_instance
(id, eval_standard_dict, pred_result_dir, gold_result_dir)
spider2-snow/evaluation_suite/evaluate.py:250
Functionevaluate_single_sql_instance
(id, eval_standard_dict, spider2sql_metadata, pred_result_dir, gold_sql_dir, gold_result_dir, temp_dir, result
spider2-snow/evaluation_suite/evaluate.py:184
Functionevaluate_spider2sql
(gold_result_dir, csv_pth, example_id, task="lite")
methods/ReFoRCE/eval.py:92
Methodexec_sql_sqlite
(self, sql_query, save_path=None, max_len=30000, sqlite_path=None)
methods/ReFoRCE/sql.py:59
Functionexecute_query
()
spider2-lite/evaluation_suite/evaluate_utils.py:162
Functionexecute_query
()
spider2-snow/evaluation_suite/evaluate_utils.py:162
Functionexecute_query
()
spider2-snow/evaluation_suite/evaluate.py:133
Functionextract_between
(file_path, start_str, end_str)
methods/ReFoRCE/utils.py:77
Functionextract_code_blocks
(text: str, tag: str)
methods/ReFoRCE/utils.py:379
Methodextract_real_table_name
(table)
methods/SL/gen_sl/naive_parser.py:26
Functionfilter_json
(raw_response: str)
spider2-lite/baselines/dailsql/utils/utils.py:174
Methodflush
(self)
spider2-lite/evaluation_suite/evaluate.py:29
Methodflush
(self)
spider2-snow/evaluation_suite/evaluate.py:34
Methodformat_example
(self, example: dict)
spider2-lite/baselines/dailsql/prompt/ExampleFormatTemplate.py:22
Methodformat_example
(self, example: dict)
spider2-lite/baselines/dailsql/prompt/ExampleFormatTemplate.py:35
Methodformat_example
(self, example: dict)
spider2-lite/baselines/dailsql/prompt/ExampleFormatTemplate.py:47
Methodformat_example
(self, example: dict)
spider2-lite/baselines/dailsql/prompt/ExampleFormatTemplate.py:59
Methodformat_example
(self, example: dict)
spider2-lite/baselines/dailsql/prompt/ExampleFormatTemplate.py:72
Methodformat_question
(self, example: dict)
spider2-lite/baselines/dailsql/prompt/PromptReprTemplate.py:135
Methodformat_question
(self, example: dict)
spider2-lite/baselines/dailsql/prompt/PromptReprTemplate.py:159
Methodformat_question
(self, example: dict)
spider2-lite/baselines/dailsql/prompt/PromptReprTemplate.py:214
Methodformat_question
(self, example: dict)
spider2-lite/baselines/dailsql/prompt/PromptReprTemplate.py:238
Methodformat_question
(self, example: dict)
spider2-lite/baselines/dailsql/prompt/PromptReprTemplate.py:273
Methodformat_question
(self, example: dict)
spider2-lite/baselines/dailsql/prompt/PromptReprTemplate.py:329
Methodformat_question
(self, example: dict, args)
spider2-lite/baselines/dailsql/prompt/PromptReprTemplate.py:358
Methodformat_question
(self, example: dict)
spider2-lite/baselines/dailsql/prompt/PromptReprTemplate.py:380
Methodformat_question
(self, example: dict)
spider2-lite/baselines/dailsql/prompt/PromptReprTemplate.py:404
Methodformat_question
(self, example: dict)
spider2-lite/baselines/dailsql/prompt/PromptReprTemplate.py:428
Methodformat_question
(self, example: dict)
spider2-lite/baselines/dailsql/prompt/PromptReprTemplate.py:556
Methodformat_target
(self, example: dict)
spider2-lite/baselines/dailsql/prompt/PromptReprTemplate.py:202
Methodformat_target
(self, example: dict)
spider2-lite/baselines/dailsql/prompt/PromptReprTemplate.py:317
Methodformat_target
(self, example: dict)
spider2-lite/baselines/dailsql/prompt/PromptReprTemplate.py:466
Methodformat_target
(self, example: dict)
spider2-lite/baselines/dailsql/prompt/PromptReprTemplate.py:490
Methodformat_target
(self, example: dict)
spider2-lite/baselines/dailsql/prompt/PromptReprTemplate.py:517
Methodformat_target
(self, example: dict)
spider2-lite/baselines/dailsql/prompt/PromptReprTemplate.py:545
Methodforward
Implements the multihead softmax attention. Arguments --------- q, k, v: The tensor containing the query, key, and value.
spider2-lite/baselines/codes/modeling_gpt_bigcode.py:119
Methodforward
( self, hidden_states: torch.Tensor, layer_past: Optional[torch.Tensor] = None,
spider2-lite/baselines/codes/modeling_gpt_bigcode.py:291
Methodforward
(self, hidden_states: Optional[Tuple[torch.Tensor]])
spider2-lite/baselines/codes/modeling_gpt_bigcode.py:372
Methodforward
( self, hidden_states: Optional[Tuple[torch.Tensor]], layer_past: Optional[torch.Tenso
spider2-lite/baselines/codes/modeling_gpt_bigcode.py:398
Methodforward
( self, input_ids: Optional[torch.Tensor] = None, past_key_values: Optional[Union[List
spider2-lite/baselines/codes/modeling_gpt_bigcode.py:635
Methodforward
r""" labels (`torch.Tensor` of shape `(batch_size, sequence_length)`, *optional*): Labels for language modeling. Note that the lab
spider2-lite/baselines/codes/modeling_gpt_bigcode.py:880
Methodforward
r""" labels (`torch.Tensor` of shape `(batch_size,)`, *optional*): Labels for computing the sequence classification/regression los
spider2-lite/baselines/codes/modeling_gpt_bigcode.py:986
Methodforward
r""" labels (`torch.Tensor` of shape `(batch_size, sequence_length)`, *optional*): Labels for computing the sequence classificatio
spider2-lite/baselines/codes/modeling_gpt_bigcode.py:1110
Methodforward
( self, encoder_input_ids, encoder_attention_mask, batch_aligned_column_info_i
spider2-lite/baselines/codes/utils/classifier_model.py:166
Methodforward
(self, input_tensor, target_tensor)
spider2-lite/baselines/codes/utils/classifier_loss.py:15
Methodget_all_json
(self)
spider2-lite/baselines/dailsql/utils/data_builder.py:160
Methodget_all_tables_bird
(self, data)
methods/SL/gen_sl/naive_parser.py:85
Functionget_bigquery_sql_result
is_save = True, output a 'result.csv' if_save = False, output a string
spider2-lite/evaluation_suite/evaluate_utils.py:153
Functionget_database_matches
( question: str, table_name: str, column_name: str, db_path: str, top_k_matches: int = 2,
spider2-lite/baselines/codes/utils/bridge_content_encoder.py:220
Functionget_db_schema
(db_path, db_comments, db_id)
spider2-lite/baselines/codes/utils/db_utils.py:138
Methodget_example_prefix
(self)
spider2-lite/baselines/dailsql/prompt/ExampleFormatTemplate.py:19
Methodget_example_prefix
(self)
spider2-lite/baselines/dailsql/prompt/ExampleFormatTemplate.py:32
Methodget_example_prefix
(self)
spider2-lite/baselines/dailsql/prompt/ExampleFormatTemplate.py:44
Methodget_example_prefix
(self)
spider2-lite/baselines/dailsql/prompt/ExampleFormatTemplate.py:56
Methodget_example_prefix
(self)
spider2-lite/baselines/dailsql/prompt/ExampleFormatTemplate.py:69
Methodget_example_quality_for_each
(self)
spider2-lite/baselines/dailsql/prompt/PromptICLTemplate.py:26
Methodget_examples
(self, target, num_example, cross_domain=False)
spider2-lite/baselines/dailsql/prompt/ExampleSelectorTemplate.py:34
Methodget_examples
(self, target, num_example, cross_domain=False)
spider2-lite/baselines/dailsql/prompt/ExampleSelectorTemplate.py:57
Methodget_examples
(self, target, num_example, cross_domain=False)
spider2-lite/baselines/dailsql/prompt/ExampleSelectorTemplate.py:92
Methodget_examples
(self, target, num_example, cross_domain=False)
spider2-lite/baselines/dailsql/prompt/ExampleSelectorTemplate.py:126
Methodget_examples
(self, target, num_example, cross_domain=False)
spider2-lite/baselines/dailsql/prompt/ExampleSelectorTemplate.py:166
Methodget_examples
(self, target, num_example, cross_domain=False)
spider2-lite/baselines/dailsql/prompt/ExampleSelectorTemplate.py:217
Methodget_examples
(self, target, num_example, cross_domain=False)
spider2-lite/baselines/dailsql/prompt/ExampleSelectorTemplate.py:251
Methodget_examples
(self, target, num_example, cross_domain=False)
spider2-lite/baselines/dailsql/prompt/ExampleSelectorTemplate.py:298
Methodget_examples
(self, target, num_example, cross_domain=False)
spider2-lite/baselines/dailsql/prompt/ExampleSelectorTemplate.py:335
Methodget_examples
(self, target, num_example, cross_domain=False)
spider2-lite/baselines/dailsql/prompt/ExampleSelectorTemplate.py:387
Methodget_input_embeddings
(self)
spider2-lite/baselines/codes/modeling_gpt_bigcode.py:623
Functionget_longest
(sql_list)
methods/ReFoRCE/utils.py:50
Methodget_lr
Compute learning rate using chainable form of the scheduler.
spider2-lite/baselines/codes/utils/lr_scheduler.py:72
Functionget_metrics
(ce_json)
methods/ReFoRCE/utils.py:384
Methodget_model_response_txt
(self, prompt)
methods/ReFoRCE/chat.py:31
Methodget_output_embeddings
(self)
spider2-lite/baselines/codes/modeling_gpt_bigcode.py:831
Methodget_path_db
(self, db_id)
spider2-lite/baselines/dailsql/utils/data_builder.py:62
Methodget_path_db
(self, db_id)
spider2-lite/baselines/dailsql/utils/data_builder.py:248
Methodget_path_sql
(self, db_id)
spider2-lite/baselines/dailsql/utils/data_builder.py:55
Methodget_prompt_convert_symbols
(self)
methods/ReFoRCE/prompt.py:44
Methodget_prompt_dialect_string_matching
(self, api)
methods/ReFoRCE/prompt.py:66
Methodget_prompt_fuzzy_query
(self)
methods/ReFoRCE/prompt.py:40
Methodget_prompt_knowledge
(self)
methods/ReFoRCE/prompt.py:46
Functionget_relevant_tables
(data_jsons, RELEVANT_TABLE_BADCASE, RELEVANT_TABLE_TOTALCASE)
spider2-lite/baselines/dailsql/utils/linking_utils/application.py:72
Methodget_response
(self, prompt)
methods/ReFoRCE/chat.py:89
← previousnext →401–500 of 554, ranked by callers