├── LICENSE ├── README.md ├── data ├── mapping │ └── reference_all_mappings.json ├── openai_input │ ├── reference_Pre_input.jsonl │ └── reference_Second_input.jsonl └── raw_format_data │ └── reference_raw_format_data.json ├── data_process ├── data_process_config.py ├── description_mapping.py ├── error_info │ └── error_index └── sql_data_process_BIRD.py ├── evaluation └── evaluation.py ├── figs ├── Ablation_Dev.png ├── Comparison.png └── GSR.png ├── output └── GSR-dev.sql ├── requirements.txt ├── run ├── GSR.py └── run_config.py └── tools ├── extractor.py ├── format_masked_regenerate_schema.py ├── similarity_search.py ├── sql_executor.py ├── tools_config.py └── value_condition_check.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GSR-SQL/GSR/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GSR-SQL/GSR/HEAD/README.md -------------------------------------------------------------------------------- /data/mapping/reference_all_mappings.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GSR-SQL/GSR/HEAD/data/mapping/reference_all_mappings.json -------------------------------------------------------------------------------- /data/openai_input/reference_Pre_input.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GSR-SQL/GSR/HEAD/data/openai_input/reference_Pre_input.jsonl -------------------------------------------------------------------------------- /data/openai_input/reference_Second_input.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GSR-SQL/GSR/HEAD/data/openai_input/reference_Second_input.jsonl -------------------------------------------------------------------------------- /data/raw_format_data/reference_raw_format_data.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GSR-SQL/GSR/HEAD/data/raw_format_data/reference_raw_format_data.json -------------------------------------------------------------------------------- /data_process/data_process_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GSR-SQL/GSR/HEAD/data_process/data_process_config.py -------------------------------------------------------------------------------- /data_process/description_mapping.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GSR-SQL/GSR/HEAD/data_process/description_mapping.py -------------------------------------------------------------------------------- /data_process/error_info/error_index: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /data_process/sql_data_process_BIRD.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GSR-SQL/GSR/HEAD/data_process/sql_data_process_BIRD.py -------------------------------------------------------------------------------- /evaluation/evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GSR-SQL/GSR/HEAD/evaluation/evaluation.py -------------------------------------------------------------------------------- /figs/Ablation_Dev.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GSR-SQL/GSR/HEAD/figs/Ablation_Dev.png -------------------------------------------------------------------------------- /figs/Comparison.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GSR-SQL/GSR/HEAD/figs/Comparison.png -------------------------------------------------------------------------------- /figs/GSR.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GSR-SQL/GSR/HEAD/figs/GSR.png -------------------------------------------------------------------------------- /output/GSR-dev.sql: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GSR-SQL/GSR/HEAD/output/GSR-dev.sql -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GSR-SQL/GSR/HEAD/requirements.txt -------------------------------------------------------------------------------- /run/GSR.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GSR-SQL/GSR/HEAD/run/GSR.py -------------------------------------------------------------------------------- /run/run_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GSR-SQL/GSR/HEAD/run/run_config.py -------------------------------------------------------------------------------- /tools/extractor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GSR-SQL/GSR/HEAD/tools/extractor.py -------------------------------------------------------------------------------- /tools/format_masked_regenerate_schema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GSR-SQL/GSR/HEAD/tools/format_masked_regenerate_schema.py -------------------------------------------------------------------------------- /tools/similarity_search.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GSR-SQL/GSR/HEAD/tools/similarity_search.py -------------------------------------------------------------------------------- /tools/sql_executor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GSR-SQL/GSR/HEAD/tools/sql_executor.py -------------------------------------------------------------------------------- /tools/tools_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GSR-SQL/GSR/HEAD/tools/tools_config.py -------------------------------------------------------------------------------- /tools/value_condition_check.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GSR-SQL/GSR/HEAD/tools/value_condition_check.py --------------------------------------------------------------------------------