MCPcopy Create free account

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

Functions554 in github.com/Snowflake-Labs/ReFoRCE

⨍ Functions 554◇ Types & classes 83↳ Endpoints 1
↓ 1 callersMethodrecord_pattern_similarity
(self, examples, target)
spider2-lite/baselines/dailsql/prompt/PromptICLTemplate.py:32
↓ 1 callersFunctionreduce_columns
(sql: str, subset_columns: set[str])
methods/ReFoRCE/schema_linking.py:33
↓ 1 callersFunctionreduce_ddl
(example_path, dictionaries, linked_json, clear_long_eg_des=False, reduce_col=False, linking_method="naive", t
methods/ReFoRCE/schema_linking.py:63
↓ 1 callersFunctionremove_declare_lines
(sql_script: str)
methods/ReFoRCE/utils.py:364
↓ 1 callersFunctionremove_distinct
(s)
spider2-lite/baselines/dailsql/utils/post_process.py:206
↓ 1 callersFunctionremove_semicolon
(s)
spider2-lite/baselines/dailsql/utils/utils.py:356
↓ 1 callersFunctionremove_table_alias
(s)
spider2-lite/baselines/dailsql/utils/utils.py:399
↓ 1 callersFunctionreplace_cur_year
(query: str)
spider2-lite/baselines/dailsql/utils/post_process.py:134
↓ 1 callersFunctionreplace_table_names
(sql_content, selected_tables_to_dbid)
spider2-lite/baselines/dailsql/preprocessed_data/spider2_preprocess.py:47
↓ 1 callersFunctionresult_eq
(result1: List[Tuple], result2: List[Tuple], order_matters: bool)
spider2-lite/baselines/dailsql/utils/post_process.py:86
↓ 1 callersFunctionrun_with_timeout
(func, timeout, *args, **kwargs)
spider2-lite/evaluation_suite/evaluate_utils.py:28
↓ 1 callersFunctionrun_with_timeout
(func, timeout, *args, **kwargs)
spider2-snow/evaluation_suite/evaluate_utils.py:28
↓ 1 callersMethodsave
(self)
spider2-lite/baselines/dailsql/utils/linking_process.py:208
↓ 1 callersFunctionsave_correct_ids_to_csv
(output_results, result_dir)
spider2-snow/evaluation_suite/evaluate.py:304
↓ 1 callersFunctionsave_to_jsonl
(folder_names, file_path)
methods/ReFoRCE/get_metadata.py:6
↓ 1 callersFunctionschema_linking_producer
(test, table, db, dataset_dir, compute_cv_link=False, args=None)
spider2-lite/baselines/dailsql/data_preprocess.py:18
↓ 1 callersFunctionschema_linking_prompt_maker
(test_sample_text, db_names, spider_schema, spider_foreign)
spider2-lite/baselines/dinsql/DIN-SQL.py:615
↓ 1 callersMethodself_correct
(self, sql, error, logger, simplify=False)
methods/ReFoRCE/agent.py:110
↓ 1 callersMethodself_refine
(self, args, logger, question, format_csv, table_struct, table_info, response_pre_txt, pre_info, csv_save_path
methods/ReFoRCE/agent.py:169
↓ 1 callersFunctionsetup_add_schema
(args)
methods/ReFoRCE/spider_agent_setup_snow.py:42
↓ 1 callersFunctionsetup_snowflake
()
methods/ReFoRCE/spider_agent_setup_snow.py:21
↓ 1 callersFunctionspider2_postprocess_single_sql
status: deprecated. we don't use self-consistency in dai-sql
spider2-lite/baselines/utils/post_utils.py:32
↓ 1 callersMethodspider2_table_back
(self, parsed)
methods/SL/gen_sl/naive_parser.py:183
↓ 1 callersFunctionsplit_sample
(sample, tokenizer)
spider2-lite/baselines/codes/schema_item_filter.py:110
↓ 1 callersFunctionsplit_schema_with_overlap
(schema: str, table_structure: str, overlap: int = 1000)
methods/SL/gen_sl/mk_prompt_spider2.py:82
↓ 1 callersFunctionsql2skeleton
(sql: str, db_schema)
spider2-lite/baselines/dailsql/utils/utils.py:457
↓ 1 callersFunctionsql_normalization
(sql)
spider2-lite/baselines/dailsql/utils/utils.py:329
↓ 1 callersMethodstart_db_sf
(self, ex_id)
methods/ReFoRCE/sql.py:43
↓ 1 callersMethodstart_db_sqlite
(self, sqlite_path)
methods/ReFoRCE/sql.py:36
↓ 1 callersMethodtable_column_cls
( self, encoder_input_ids, encoder_input_attention_mask, batch_aligned_column_
spider2-lite/baselines/codes/utils/classifier_model.py:90
↓ 1 callersMethodtable_column_cross_attention
( self, table_name_embeddings_in_one_db, column_info_embeddings_in_one_db, c
spider2-lite/baselines/codes/utils/classifier_model.py:62
↓ 1 callersFunctiontext2sql_func
(model, inputs, tokenizer, max_new_tokens)
spider2-lite/baselines/codes/text2sql_zero_shot.py:32
↓ 1 callersFunctiontext2sql_func
(model, text2sql_input_seq, tokenizer, max_tokens, max_new_tokens, eos_token_id)
spider2-lite/baselines/codes/text2sql_few_shot.py:122
↓ 1 callersFunctionto_markdown
(sample_rows)
spider2-lite/baselines/dailsql/utils/utils.py:211
↓ 1 callersMethodtokenize
(self, text)
spider2-lite/baselines/dailsql/utils/pretrained_embeddings.py:76
↓ 1 callersMethodtokenize_for_copying
(self, text)
spider2-lite/baselines/dailsql/utils/pretrained_embeddings.py:84
↓ 1 callersFunctionupcast_masked_softmax
( x: torch.Tensor, mask: torch.Tensor, mask_value: torch.Tensor, scale: float, softmax_dtype: torch.dtype
spider2-lite/baselines/codes/modeling_gpt_bigcode.py:74
↓ 1 callersFunctionupcast_softmax
(x: torch.Tensor, scale: float, softmax_dtype: torch.dtype)
spider2-lite/baselines/codes/modeling_gpt_bigcode.py:85
↓ 1 callersFunctionupdate_results
(gold_result_dir)
methods/ReFoRCE/eval.py:133
↓ 1 callersMethodvalidate_item
(self, item, schema, section)
spider2-lite/baselines/dailsql/utils/linking_process.py:145
↓ 1 callersMethodvote_result
(self, search_directory, args, sql_paths, table_info, task)
methods/ReFoRCE/agent.py:334
↓ 1 callersFunctionwhite_space_fix
(s)
spider2-lite/baselines/dailsql/utils/utils.py:331
Method__del__
(self)
spider2-lite/baselines/dailsql/utils/linking_utils/corenlp.py:44
Method__getitem__
(self, index)
spider2-lite/baselines/codes/utils/load_sft_dataset.py:127
Method__getitem__
(self, index)
spider2-lite/baselines/codes/utils/load_pt_dataset.py:57
Method__getitem__
(self, index)
spider2-lite/baselines/codes/utils/load_classifier_dataset.py:41
Method__init__
(self, causal=False, softmax_scale=None, attention_dropout=0.0, device=None, dtype=None)
spider2-lite/baselines/codes/modeling_gpt_bigcode.py:109
Method__init__
(self, config, is_cross_attention=False, layer_idx=None)
spider2-lite/baselines/codes/modeling_gpt_bigcode.py:161
Method__init__
(self, config, layer_idx=None)
spider2-lite/baselines/codes/modeling_gpt_bigcode.py:381
Method__init__
(self, *inputs, **kwargs)
spider2-lite/baselines/codes/modeling_gpt_bigcode.py:473
Method__init__
(self, config)
spider2-lite/baselines/codes/modeling_gpt_bigcode.py:601
Method__init__
(self, config)
spider2-lite/baselines/codes/modeling_gpt_bigcode.py:823
Method__init__
(self, config)
spider2-lite/baselines/codes/modeling_gpt_bigcode.py:976
Method__init__
(self, config)
spider2-lite/baselines/codes/modeling_gpt_bigcode.py:1092
Method__init__
(self, model_save_path)
spider2-lite/baselines/codes/schema_item_filter.py:276
Method__init__
(self, model_name_or_path, mode)
spider2-lite/baselines/codes/utils/classifier_model.py:7
Method__init__
(self, text2sql_data_dir, tokenizer, max_tokens, mode, table_num, column_num, sic_path, args)
spider2-lite/baselines/codes/utils/load_sft_dataset.py:101
Method__init__
(self, pt_data_dir, block_size)
spider2-lite/baselines/codes/utils/load_pt_dataset.py:50
Method__init__
(self, start: int, size: int)
spider2-lite/baselines/codes/utils/bridge_content_encoder.py:58
Method__init__
(self, dataset_dir)
spider2-lite/baselines/codes/utils/load_classifier_dataset.py:6
Method__init__
Args: optimizer (Optimizer): Wrapped optimizer. warmup_epochs (int): Maximum number of iterations for linear warmup
spider2-lite/baselines/codes/utils/lr_scheduler.py:47
Method__init__
(self, weight=None, gamma=0.5, reduction=None)
spider2-lite/baselines/codes/utils/classifier_loss.py:8
Method__init__
(self, alpha, gamma)
spider2-lite/baselines/codes/utils/classifier_loss.py:31
Method__init__
(self, path_data, args)
spider2-lite/baselines/dailsql/utils/data_builder.py:234
Method__init__
(self, kind, lemmatize=False)
spider2-lite/baselines/dailsql/utils/pretrained_embeddings.py:45
Method__init__
( self, save_path, min_freq=3, max_count=5000, inc
spider2-lite/baselines/dailsql/utils/linking_process.py:110
Method__init__
(self)
spider2-lite/baselines/dailsql/utils/linking_utils/corenlp.py:31
Method__init__
(self, data, *args, **kwargs)
spider2-lite/baselines/dailsql/prompt/ExampleSelectorTemplate.py:30
Method__init__
(self, data, *args, **kwargs)
spider2-lite/baselines/dailsql/prompt/ExampleSelectorTemplate.py:46
Method__init__
(self, data, *args, **kwargs)
spider2-lite/baselines/dailsql/prompt/ExampleSelectorTemplate.py:83
Method__init__
(self, data, *args, **kwargs)
spider2-lite/baselines/dailsql/prompt/ExampleSelectorTemplate.py:115
Method__init__
(self, data, *args, **kwargs)
spider2-lite/baselines/dailsql/prompt/ExampleSelectorTemplate.py:153
Method__init__
(self, data, *args, **kwargs)
spider2-lite/baselines/dailsql/prompt/ExampleSelectorTemplate.py:205
Method__init__
(self, data, *args, **kwargs)
spider2-lite/baselines/dailsql/prompt/ExampleSelectorTemplate.py:241
Method__init__
(self, data, *args, **kwargs)
spider2-lite/baselines/dailsql/prompt/ExampleSelectorTemplate.py:289
Method__init__
(self, data, *args, **kwargs)
spider2-lite/baselines/dailsql/prompt/ExampleSelectorTemplate.py:322
Method__init__
(self, data, *args, **kwargs)
spider2-lite/baselines/dailsql/prompt/ExampleSelectorTemplate.py:374
Method__init__
(self, *args, **kwargs)
spider2-lite/baselines/dailsql/prompt/PromptReprTemplate.py:12
Method__init__
(self, *args, **kwargs)
spider2-lite/baselines/dailsql/prompt/prompt_builder.py:106
Method__init__
(self, tokenizer: str, *args, **kwargs)
spider2-lite/baselines/dailsql/prompt/PromptICLTemplate.py:9
Method__init__
(self, filename)
spider2-lite/evaluation_suite/evaluate.py:21
Method__init__
(self, filename)
spider2-snow/evaluation_suite/evaluate.py:24
Method__init__
(self)
methods/SL/gen_sl/naive_parser.py:15
Method__init__
(self)
methods/ReFoRCE/prompt.py:29
Method__init__
(self, db_path, sql_data, search_directory, prompt_class: Type[Prompts], sql_env: Type[SqlEnv]=None, chat_sess
methods/ReFoRCE/agent.py:15
Method__init__
(self, model: str, temperature: float = 1.0)
methods/ReFoRCE/chat.py:6
Method__init__
(self, azure=False, model="gpt-4o", temperature=1.0)
methods/ReFoRCE/chat.py:59
Method__init__
(self)
methods/ReFoRCE/sql.py:13
Method__len__
(self)
spider2-lite/baselines/codes/utils/load_sft_dataset.py:140
Method__len__
(self)
spider2-lite/baselines/codes/utils/load_pt_dataset.py:65
Method__len__
(self)
spider2-lite/baselines/codes/utils/load_classifier_dataset.py:38
Method_get_closed_form_lr
Called when epoch is passed as a param to the `step` function of the scheduler.
spider2-lite/baselines/codes/utils/lr_scheduler.py:109
Method_init_weights
Initialize the weights.
spider2-lite/baselines/codes/modeling_gpt_bigcode.py:476
Method_reorder_cache
This function is used to re-order the `past_key_values` cache if [`~PreTrainedModel.beam_search`] or [`~PreTrainedModel.beam_sample`]
spider2-lite/baselines/codes/modeling_gpt_bigcode.py:947
Method_set_gradient_checkpointing
(self, module, value=False)
spider2-lite/baselines/codes/modeling_gpt_bigcode.py:504
Method_tokenize
(self, presplit, unsplit)
spider2-lite/baselines/dailsql/utils/linking_process.py:198
Methodadd_item
Add an item to be preprocessed.
spider2-lite/baselines/dailsql/utils/linking_utils/abstract_preproc.py:24
Methodadd_sqls_from_json
(self, sqls, JS_PTHS, example_id)
methods/SL/gen_sl/naive_parser.py:115
Functionannotate
(text, annotators=None, output_format=None, properties=None)
spider2-lite/baselines/dailsql/utils/linking_utils/corenlp.py:63
Functionappend_json_line
(path, obj)
methods/SL/gen_sl/utils.py:27
← previousnext →301–400 of 554, ranked by callers