├── LICENSE ├── README.md ├── data ├── annotated_qdmr │ ├── imdb_qdmr.csv │ └── yelp_qdmr.csv ├── sql_synthesis_results │ ├── gold_qdmr_supervision │ │ ├── academic.csv │ │ ├── geo880.csv │ │ ├── gold_qdmr_synthesized_full.zip │ │ ├── imdb.csv │ │ ├── spider_dev.csv │ │ ├── spider_train.csv │ │ └── yelp.csv │ ├── grounding_statistics.txt │ ├── predicted_qdmr_supervision │ │ ├── geo880_dev_test_pred.csv │ │ ├── geo880_dev_test_pred.json │ │ ├── spider_dev_pred.csv │ │ └── spider_dev_pred.json │ └── sql_synthesis_input_example.csv └── text_to_sql │ ├── encoded_qdmr_datasets.zip │ └── gold_sql_datasets.zip ├── requirements.txt ├── requirements_qdmr_parser.txt └── src ├── data_generation ├── db_schema.py ├── graph_utils.py ├── ground_example.py ├── grounded_qdmr.py ├── grounding_repairs.py ├── main.py ├── operator_identifier.py ├── predicted_sql.py ├── preprocess_db.py ├── preprocess_grounding_data.py ├── qdmr_editor.py ├── qdmr_encoding.py ├── qdmr_encoding_parser.py ├── qdmr_grounding.py ├── qdmr_identifier.py ├── schema_parser.py ├── sql_execution.py ├── sql_parser.py ├── sql_query.py ├── test_encoding_conversion.py ├── test_grounded_qdmr.py ├── test_predicted_grounded_qdmr.py ├── utils.py ├── write_encoding.py └── write_grounding.py ├── qdmr_parser ├── dataset_qdmr.py ├── eval_qdmr │ ├── eval_string_match.py │ └── sari_hook.py ├── model.py ├── test.py ├── train.py └── utils_data.py └── text_to_sql ├── dataset_qdmr.py ├── dataset_spider.py ├── dataset_utils.py ├── eval_exec ├── db_schema.py ├── graph_utils.py ├── grounded_qdmr.py ├── operator_identifier.py ├── predicted_sql.py ├── preprocess_db.py ├── qdmr_encoding.py ├── qdmr_encoding_parser.py ├── qdmr_grounding.py ├── qdmr_identifier.py ├── qdmr_sql.py ├── schema_parser.py ├── sql_execution.py ├── sql_parser.py └── utils.py ├── eval_spider.py ├── evaluation.py ├── model.py ├── process_sql.py ├── test.py ├── test_exec_eval.py └── train.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tomerwolgithub/question-decomposition-to-sql/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tomerwolgithub/question-decomposition-to-sql/HEAD/README.md -------------------------------------------------------------------------------- /data/annotated_qdmr/imdb_qdmr.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tomerwolgithub/question-decomposition-to-sql/HEAD/data/annotated_qdmr/imdb_qdmr.csv -------------------------------------------------------------------------------- /data/annotated_qdmr/yelp_qdmr.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tomerwolgithub/question-decomposition-to-sql/HEAD/data/annotated_qdmr/yelp_qdmr.csv -------------------------------------------------------------------------------- /data/sql_synthesis_results/gold_qdmr_supervision/academic.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tomerwolgithub/question-decomposition-to-sql/HEAD/data/sql_synthesis_results/gold_qdmr_supervision/academic.csv -------------------------------------------------------------------------------- /data/sql_synthesis_results/gold_qdmr_supervision/geo880.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tomerwolgithub/question-decomposition-to-sql/HEAD/data/sql_synthesis_results/gold_qdmr_supervision/geo880.csv -------------------------------------------------------------------------------- /data/sql_synthesis_results/gold_qdmr_supervision/gold_qdmr_synthesized_full.zip: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tomerwolgithub/question-decomposition-to-sql/HEAD/data/sql_synthesis_results/gold_qdmr_supervision/gold_qdmr_synthesized_full.zip -------------------------------------------------------------------------------- /data/sql_synthesis_results/gold_qdmr_supervision/imdb.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tomerwolgithub/question-decomposition-to-sql/HEAD/data/sql_synthesis_results/gold_qdmr_supervision/imdb.csv -------------------------------------------------------------------------------- /data/sql_synthesis_results/gold_qdmr_supervision/spider_dev.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tomerwolgithub/question-decomposition-to-sql/HEAD/data/sql_synthesis_results/gold_qdmr_supervision/spider_dev.csv -------------------------------------------------------------------------------- /data/sql_synthesis_results/gold_qdmr_supervision/spider_train.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tomerwolgithub/question-decomposition-to-sql/HEAD/data/sql_synthesis_results/gold_qdmr_supervision/spider_train.csv -------------------------------------------------------------------------------- /data/sql_synthesis_results/gold_qdmr_supervision/yelp.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tomerwolgithub/question-decomposition-to-sql/HEAD/data/sql_synthesis_results/gold_qdmr_supervision/yelp.csv -------------------------------------------------------------------------------- /data/sql_synthesis_results/grounding_statistics.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tomerwolgithub/question-decomposition-to-sql/HEAD/data/sql_synthesis_results/grounding_statistics.txt -------------------------------------------------------------------------------- /data/sql_synthesis_results/predicted_qdmr_supervision/geo880_dev_test_pred.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tomerwolgithub/question-decomposition-to-sql/HEAD/data/sql_synthesis_results/predicted_qdmr_supervision/geo880_dev_test_pred.csv -------------------------------------------------------------------------------- /data/sql_synthesis_results/predicted_qdmr_supervision/geo880_dev_test_pred.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tomerwolgithub/question-decomposition-to-sql/HEAD/data/sql_synthesis_results/predicted_qdmr_supervision/geo880_dev_test_pred.json -------------------------------------------------------------------------------- /data/sql_synthesis_results/predicted_qdmr_supervision/spider_dev_pred.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tomerwolgithub/question-decomposition-to-sql/HEAD/data/sql_synthesis_results/predicted_qdmr_supervision/spider_dev_pred.csv -------------------------------------------------------------------------------- /data/sql_synthesis_results/predicted_qdmr_supervision/spider_dev_pred.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tomerwolgithub/question-decomposition-to-sql/HEAD/data/sql_synthesis_results/predicted_qdmr_supervision/spider_dev_pred.json -------------------------------------------------------------------------------- /data/sql_synthesis_results/sql_synthesis_input_example.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tomerwolgithub/question-decomposition-to-sql/HEAD/data/sql_synthesis_results/sql_synthesis_input_example.csv -------------------------------------------------------------------------------- /data/text_to_sql/encoded_qdmr_datasets.zip: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tomerwolgithub/question-decomposition-to-sql/HEAD/data/text_to_sql/encoded_qdmr_datasets.zip -------------------------------------------------------------------------------- /data/text_to_sql/gold_sql_datasets.zip: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tomerwolgithub/question-decomposition-to-sql/HEAD/data/text_to_sql/gold_sql_datasets.zip -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tomerwolgithub/question-decomposition-to-sql/HEAD/requirements.txt -------------------------------------------------------------------------------- /requirements_qdmr_parser.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tomerwolgithub/question-decomposition-to-sql/HEAD/requirements_qdmr_parser.txt -------------------------------------------------------------------------------- /src/data_generation/db_schema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tomerwolgithub/question-decomposition-to-sql/HEAD/src/data_generation/db_schema.py -------------------------------------------------------------------------------- /src/data_generation/graph_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tomerwolgithub/question-decomposition-to-sql/HEAD/src/data_generation/graph_utils.py -------------------------------------------------------------------------------- /src/data_generation/ground_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tomerwolgithub/question-decomposition-to-sql/HEAD/src/data_generation/ground_example.py -------------------------------------------------------------------------------- /src/data_generation/grounded_qdmr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tomerwolgithub/question-decomposition-to-sql/HEAD/src/data_generation/grounded_qdmr.py -------------------------------------------------------------------------------- /src/data_generation/grounding_repairs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tomerwolgithub/question-decomposition-to-sql/HEAD/src/data_generation/grounding_repairs.py -------------------------------------------------------------------------------- /src/data_generation/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tomerwolgithub/question-decomposition-to-sql/HEAD/src/data_generation/main.py -------------------------------------------------------------------------------- /src/data_generation/operator_identifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tomerwolgithub/question-decomposition-to-sql/HEAD/src/data_generation/operator_identifier.py -------------------------------------------------------------------------------- /src/data_generation/predicted_sql.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tomerwolgithub/question-decomposition-to-sql/HEAD/src/data_generation/predicted_sql.py -------------------------------------------------------------------------------- /src/data_generation/preprocess_db.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tomerwolgithub/question-decomposition-to-sql/HEAD/src/data_generation/preprocess_db.py -------------------------------------------------------------------------------- /src/data_generation/preprocess_grounding_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tomerwolgithub/question-decomposition-to-sql/HEAD/src/data_generation/preprocess_grounding_data.py -------------------------------------------------------------------------------- /src/data_generation/qdmr_editor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tomerwolgithub/question-decomposition-to-sql/HEAD/src/data_generation/qdmr_editor.py -------------------------------------------------------------------------------- /src/data_generation/qdmr_encoding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tomerwolgithub/question-decomposition-to-sql/HEAD/src/data_generation/qdmr_encoding.py -------------------------------------------------------------------------------- /src/data_generation/qdmr_encoding_parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tomerwolgithub/question-decomposition-to-sql/HEAD/src/data_generation/qdmr_encoding_parser.py -------------------------------------------------------------------------------- /src/data_generation/qdmr_grounding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tomerwolgithub/question-decomposition-to-sql/HEAD/src/data_generation/qdmr_grounding.py -------------------------------------------------------------------------------- /src/data_generation/qdmr_identifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tomerwolgithub/question-decomposition-to-sql/HEAD/src/data_generation/qdmr_identifier.py -------------------------------------------------------------------------------- /src/data_generation/schema_parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tomerwolgithub/question-decomposition-to-sql/HEAD/src/data_generation/schema_parser.py -------------------------------------------------------------------------------- /src/data_generation/sql_execution.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tomerwolgithub/question-decomposition-to-sql/HEAD/src/data_generation/sql_execution.py -------------------------------------------------------------------------------- /src/data_generation/sql_parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tomerwolgithub/question-decomposition-to-sql/HEAD/src/data_generation/sql_parser.py -------------------------------------------------------------------------------- /src/data_generation/sql_query.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tomerwolgithub/question-decomposition-to-sql/HEAD/src/data_generation/sql_query.py -------------------------------------------------------------------------------- /src/data_generation/test_encoding_conversion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tomerwolgithub/question-decomposition-to-sql/HEAD/src/data_generation/test_encoding_conversion.py -------------------------------------------------------------------------------- /src/data_generation/test_grounded_qdmr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tomerwolgithub/question-decomposition-to-sql/HEAD/src/data_generation/test_grounded_qdmr.py -------------------------------------------------------------------------------- /src/data_generation/test_predicted_grounded_qdmr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tomerwolgithub/question-decomposition-to-sql/HEAD/src/data_generation/test_predicted_grounded_qdmr.py -------------------------------------------------------------------------------- /src/data_generation/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tomerwolgithub/question-decomposition-to-sql/HEAD/src/data_generation/utils.py -------------------------------------------------------------------------------- /src/data_generation/write_encoding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tomerwolgithub/question-decomposition-to-sql/HEAD/src/data_generation/write_encoding.py -------------------------------------------------------------------------------- /src/data_generation/write_grounding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tomerwolgithub/question-decomposition-to-sql/HEAD/src/data_generation/write_grounding.py -------------------------------------------------------------------------------- /src/qdmr_parser/dataset_qdmr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tomerwolgithub/question-decomposition-to-sql/HEAD/src/qdmr_parser/dataset_qdmr.py -------------------------------------------------------------------------------- /src/qdmr_parser/eval_qdmr/eval_string_match.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tomerwolgithub/question-decomposition-to-sql/HEAD/src/qdmr_parser/eval_qdmr/eval_string_match.py -------------------------------------------------------------------------------- /src/qdmr_parser/eval_qdmr/sari_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tomerwolgithub/question-decomposition-to-sql/HEAD/src/qdmr_parser/eval_qdmr/sari_hook.py -------------------------------------------------------------------------------- /src/qdmr_parser/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tomerwolgithub/question-decomposition-to-sql/HEAD/src/qdmr_parser/model.py -------------------------------------------------------------------------------- /src/qdmr_parser/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tomerwolgithub/question-decomposition-to-sql/HEAD/src/qdmr_parser/test.py -------------------------------------------------------------------------------- /src/qdmr_parser/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tomerwolgithub/question-decomposition-to-sql/HEAD/src/qdmr_parser/train.py -------------------------------------------------------------------------------- /src/qdmr_parser/utils_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tomerwolgithub/question-decomposition-to-sql/HEAD/src/qdmr_parser/utils_data.py -------------------------------------------------------------------------------- /src/text_to_sql/dataset_qdmr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tomerwolgithub/question-decomposition-to-sql/HEAD/src/text_to_sql/dataset_qdmr.py -------------------------------------------------------------------------------- /src/text_to_sql/dataset_spider.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tomerwolgithub/question-decomposition-to-sql/HEAD/src/text_to_sql/dataset_spider.py -------------------------------------------------------------------------------- /src/text_to_sql/dataset_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tomerwolgithub/question-decomposition-to-sql/HEAD/src/text_to_sql/dataset_utils.py -------------------------------------------------------------------------------- /src/text_to_sql/eval_exec/db_schema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tomerwolgithub/question-decomposition-to-sql/HEAD/src/text_to_sql/eval_exec/db_schema.py -------------------------------------------------------------------------------- /src/text_to_sql/eval_exec/graph_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tomerwolgithub/question-decomposition-to-sql/HEAD/src/text_to_sql/eval_exec/graph_utils.py -------------------------------------------------------------------------------- /src/text_to_sql/eval_exec/grounded_qdmr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tomerwolgithub/question-decomposition-to-sql/HEAD/src/text_to_sql/eval_exec/grounded_qdmr.py -------------------------------------------------------------------------------- /src/text_to_sql/eval_exec/operator_identifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tomerwolgithub/question-decomposition-to-sql/HEAD/src/text_to_sql/eval_exec/operator_identifier.py -------------------------------------------------------------------------------- /src/text_to_sql/eval_exec/predicted_sql.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tomerwolgithub/question-decomposition-to-sql/HEAD/src/text_to_sql/eval_exec/predicted_sql.py -------------------------------------------------------------------------------- /src/text_to_sql/eval_exec/preprocess_db.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tomerwolgithub/question-decomposition-to-sql/HEAD/src/text_to_sql/eval_exec/preprocess_db.py -------------------------------------------------------------------------------- /src/text_to_sql/eval_exec/qdmr_encoding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tomerwolgithub/question-decomposition-to-sql/HEAD/src/text_to_sql/eval_exec/qdmr_encoding.py -------------------------------------------------------------------------------- /src/text_to_sql/eval_exec/qdmr_encoding_parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tomerwolgithub/question-decomposition-to-sql/HEAD/src/text_to_sql/eval_exec/qdmr_encoding_parser.py -------------------------------------------------------------------------------- /src/text_to_sql/eval_exec/qdmr_grounding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tomerwolgithub/question-decomposition-to-sql/HEAD/src/text_to_sql/eval_exec/qdmr_grounding.py -------------------------------------------------------------------------------- /src/text_to_sql/eval_exec/qdmr_identifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tomerwolgithub/question-decomposition-to-sql/HEAD/src/text_to_sql/eval_exec/qdmr_identifier.py -------------------------------------------------------------------------------- /src/text_to_sql/eval_exec/qdmr_sql.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tomerwolgithub/question-decomposition-to-sql/HEAD/src/text_to_sql/eval_exec/qdmr_sql.py -------------------------------------------------------------------------------- /src/text_to_sql/eval_exec/schema_parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tomerwolgithub/question-decomposition-to-sql/HEAD/src/text_to_sql/eval_exec/schema_parser.py -------------------------------------------------------------------------------- /src/text_to_sql/eval_exec/sql_execution.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tomerwolgithub/question-decomposition-to-sql/HEAD/src/text_to_sql/eval_exec/sql_execution.py -------------------------------------------------------------------------------- /src/text_to_sql/eval_exec/sql_parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tomerwolgithub/question-decomposition-to-sql/HEAD/src/text_to_sql/eval_exec/sql_parser.py -------------------------------------------------------------------------------- /src/text_to_sql/eval_exec/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tomerwolgithub/question-decomposition-to-sql/HEAD/src/text_to_sql/eval_exec/utils.py -------------------------------------------------------------------------------- /src/text_to_sql/eval_spider.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tomerwolgithub/question-decomposition-to-sql/HEAD/src/text_to_sql/eval_spider.py -------------------------------------------------------------------------------- /src/text_to_sql/evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tomerwolgithub/question-decomposition-to-sql/HEAD/src/text_to_sql/evaluation.py -------------------------------------------------------------------------------- /src/text_to_sql/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tomerwolgithub/question-decomposition-to-sql/HEAD/src/text_to_sql/model.py -------------------------------------------------------------------------------- /src/text_to_sql/process_sql.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tomerwolgithub/question-decomposition-to-sql/HEAD/src/text_to_sql/process_sql.py -------------------------------------------------------------------------------- /src/text_to_sql/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tomerwolgithub/question-decomposition-to-sql/HEAD/src/text_to_sql/test.py -------------------------------------------------------------------------------- /src/text_to_sql/test_exec_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tomerwolgithub/question-decomposition-to-sql/HEAD/src/text_to_sql/test_exec_eval.py -------------------------------------------------------------------------------- /src/text_to_sql/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tomerwolgithub/question-decomposition-to-sql/HEAD/src/text_to_sql/train.py --------------------------------------------------------------------------------