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
Functionget_shortest
(sql_list)
methods/ReFoRCE/utils.py:55
Functionget_snowflake_sql_result
is_save = True, output a 'result.csv' if_save = False, output a string
spider2-lite/evaluation_suite/evaluate_utils.py:125
Functionget_sql_for_database
(path_db=None, cur=None)
spider2-lite/baselines/dailsql/utils/utils.py:190
Functionget_sqlite_result
(db_path, query, save_dir=None, file_name="result.csv", chunksize=500)
spider2-lite/evaluation_suite/evaluate_utils.py:189
Methodget_tables
(self, db_id)
spider2-lite/baselines/dailsql/utils/data_builder.py:251
Methodget_test_answers
(self, mini_set=False)
spider2-lite/baselines/dailsql/utils/data_builder.py:168
Methodget_test_questions
(self, mini_set=False)
spider2-lite/baselines/dailsql/utils/data_builder.py:75
Methodget_train_answers
(self)
spider2-lite/baselines/dailsql/utils/data_builder.py:163
Methodget_train_duplicated_index
(self)
spider2-lite/baselines/dailsql/utils/data_builder.py:176
Functionhard_cut
(str_e, length=0)
methods/SL/gen_sl/utils.py:34
Functionis_number
(s)
spider2-lite/baselines/codes/utils/db_utils.py:50
Functionis_number
(s: str)
spider2-lite/baselines/codes/utils/bridge_content_encoder.py:37
Functionload_json_list_to_dict
(json_file_path)
spider2-lite/evaluation_suite/evaluate_utils.py:53
Functionload_json_list_to_dict
(json_file_path)
spider2-lite/evaluation_suite/evaluate.py:53
Functionload_json_list_to_dict
(json_file_path)
spider2-snow/evaluation_suite/evaluate_utils.py:53
Functionload_json_list_to_dict
(json_file_path)
spider2-snow/evaluation_suite/evaluate.py:59
Functionload_json_list_to_dict
(json_file_path)
methods/ReFoRCE/eval.py:24
Functionload_jsonl_to_dict
(jsonl_file)
spider2-lite/evaluation_suite/evaluate_utils.py:44
Methodlookup
Given a token, return a vector embedding if token is in vocabulary. If token is not in the vocabulary, then return None.
spider2-lite/baselines/dailsql/utils/pretrained_embeddings.py:28
Methodlookup
(self, token)
spider2-lite/baselines/dailsql/utils/pretrained_embeddings.py:96
Functionmasked_softmax
(x: torch.Tensor, mask: torch.Tensor, mask_value: torch.Tensor)
spider2-lite/baselines/codes/modeling_gpt_bigcode.py:93
Functionmatching_at_same_position
(s1, s2)
methods/ReFoRCE/utils.py:182
Functionmulti_processing_process_row
(index, row, args, output_dir, spider_schema, spider_primary, spider_foreign)
spider2-lite/baselines/dinsql/DIN-SQL.py:785
Functionparse_des
(pre_col_values, nouns, debug)
methods/SL/gen_sl/utils.py:56
Functionpostprocess_original_name
(s: str)
spider2-lite/baselines/dailsql/utils/datasets/spider.py:95
Methodprepare_inputs_for_generation
(self, input_ids, past_key_values=None, inputs_embeds=None, **kwargs)
spider2-lite/baselines/codes/modeling_gpt_bigcode.py:837
Functionprocess_batch
Function to process each batch in parallel.
spider2-lite/baselines/dailsql/ask_llm.py:16
Functionprocess_example
(ex_id)
methods/ReFoRCE/schema_linking.py:183
Functionprocess_item
(item)
methods/SL/gen_sl/infer_api.py:32
Functionprocess_question
(question_json, args, cross_domain)
spider2-lite/baselines/dailsql/generate_question.py:23
Functionprocess_sql_data
(sql_data)
methods/ReFoRCE/run.py:86
Methodretrieve_index
(self, indexes: list, db_id)
spider2-lite/baselines/dailsql/prompt/ExampleSelectorTemplate.py:23
Methodsave
Marks that all of the items have been preprocessed. Save state to disk. Used in preprocess.py, after reading all of the data.
spider2-lite/baselines/dailsql/utils/linking_utils/abstract_preproc.py:34
Methodset_input_embeddings
(self, new_embeddings)
spider2-lite/baselines/codes/modeling_gpt_bigcode.py:626
Methodset_mini_test
(self, mini_file)
spider2-lite/baselines/dailsql/utils/data_builder.py:33
Methodset_output_embeddings
(self, new_embeddings)
spider2-lite/baselines/codes/modeling_gpt_bigcode.py:834
Functionsetup_add_schema
(args)
methods/ReFoRCE/spider_agent_setup_lite.py:61
Functionsetup_bigquery
()
methods/ReFoRCE/spider_agent_setup_lite.py:41
Functionsetup_snowflake
()
methods/ReFoRCE/spider_agent_setup_lite.py:23
Functiontimeout_handler
(signum, frame)
spider2-lite/evaluation_suite/evaluate_utils.py:25
Functiontimeout_handler
(signum, frame)
spider2-snow/evaluation_suite/evaluate_utils.py:25
Methodto
Transfer the pretrained embeddings to the given device.
spider2-lite/baselines/dailsql/utils/pretrained_embeddings.py:38
Functionto_dict_with_set_values
(d)
spider2-lite/baselines/dailsql/utils/linking_utils/serialization.py:5
Functionto_dict_with_sorted_values
(d, key=None)
spider2-lite/baselines/dailsql/utils/linking_utils/serialization.py:1
Methodtokenize
Given a string, return a list of tokens suitable for lookup.
spider2-lite/baselines/dailsql/utils/pretrained_embeddings.py:18
Methodtokenize_by_coreNLP
(self, text)
spider2-lite/baselines/dailsql/utils/pretrained_embeddings.py:58
Methodtokenize_for_copying_by_coreNLP
(self, text)
spider2-lite/baselines/dailsql/utils/pretrained_embeddings.py:66
Functiontrim_block
(match)
methods/ReFoRCE/utils.py:241
Functiontuplify
(x)
spider2-lite/baselines/dailsql/utils/linking_utils/serialization.py:18
Methoduntokenize
Undo tokenize.
spider2-lite/baselines/dailsql/utils/pretrained_embeddings.py:23
Methoduntokenize
(self, tokens)
spider2-lite/baselines/dailsql/utils/pretrained_embeddings.py:93
Methodupdate_foreign_keys_by_table
(self, parsed, schema_json)
methods/SL/gen_sl/naive_parser.py:198
Methodvalidate_item
Checks whether item can be successfully preprocessed. Returns a boolean and an arbitrary object.
spider2-lite/baselines/dailsql/utils/linking_utils/abstract_preproc.py:17
Methodwrite
(self, message)
spider2-snow/evaluation_suite/evaluate.py:29
← previous501–554 of 554, ranked by callers