├── .gitignore ├── LICENSE ├── README.md ├── asdl ├── action_info.py ├── asdl.py ├── asdl_ast.py ├── decode_hypothesis.py ├── hypothesis.py ├── sql │ ├── grammar │ │ ├── sql_asdl_v0.txt │ │ ├── sql_asdl_v1.txt │ │ └── sql_asdl_v2.txt │ ├── parser │ │ ├── parser_base.py │ │ ├── parser_v0.py │ │ ├── parser_v1.py │ │ └── parser_v2.py │ ├── sql_transition_system.py │ └── unparser │ │ ├── unparser_base.py │ │ ├── unparser_v0.py │ │ ├── unparser_v1.py │ │ └── unparser_v2.py └── transition_system.py ├── eval.py ├── evaluation.py ├── model ├── decoder │ ├── onlstm.py │ └── sql_parser.py ├── encoder │ ├── functions.py │ ├── graph_encoder.py │ ├── graph_input.py │ ├── graph_output.py │ ├── lgesql.py │ └── rgatsql.py ├── model_constructor.py └── model_utils.py ├── preprocess ├── README.md ├── build_glove_vocab.py ├── common_utils.py ├── get_tables.py ├── graph_utils.py ├── parse_raw_json.py ├── parse_sql_one.py ├── process_dataset.py └── process_graphs.py ├── process_sql.py ├── requirements.txt ├── results ├── results_bert.txt ├── results_electra.txt └── results_glove.txt ├── run ├── run_evaluation.sh ├── run_lgesql_glove.sh ├── run_lgesql_plm.sh ├── run_preprocessing.sh └── run_submission.sh ├── scripts └── text2sql.py ├── setup.sh └── utils ├── args.py ├── batch.py ├── constants.py ├── evaluator.py ├── example.py ├── graph_example.py ├── hyperparams.py ├── initialization.py ├── optimization.py ├── vocab.py └── word2vec.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/X-LANCE/text2sql-lgesql/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/X-LANCE/text2sql-lgesql/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/X-LANCE/text2sql-lgesql/HEAD/README.md -------------------------------------------------------------------------------- /asdl/action_info.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/X-LANCE/text2sql-lgesql/HEAD/asdl/action_info.py -------------------------------------------------------------------------------- /asdl/asdl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/X-LANCE/text2sql-lgesql/HEAD/asdl/asdl.py -------------------------------------------------------------------------------- /asdl/asdl_ast.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/X-LANCE/text2sql-lgesql/HEAD/asdl/asdl_ast.py -------------------------------------------------------------------------------- /asdl/decode_hypothesis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/X-LANCE/text2sql-lgesql/HEAD/asdl/decode_hypothesis.py -------------------------------------------------------------------------------- /asdl/hypothesis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/X-LANCE/text2sql-lgesql/HEAD/asdl/hypothesis.py -------------------------------------------------------------------------------- /asdl/sql/grammar/sql_asdl_v0.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/X-LANCE/text2sql-lgesql/HEAD/asdl/sql/grammar/sql_asdl_v0.txt -------------------------------------------------------------------------------- /asdl/sql/grammar/sql_asdl_v1.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/X-LANCE/text2sql-lgesql/HEAD/asdl/sql/grammar/sql_asdl_v1.txt -------------------------------------------------------------------------------- /asdl/sql/grammar/sql_asdl_v2.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/X-LANCE/text2sql-lgesql/HEAD/asdl/sql/grammar/sql_asdl_v2.txt -------------------------------------------------------------------------------- /asdl/sql/parser/parser_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/X-LANCE/text2sql-lgesql/HEAD/asdl/sql/parser/parser_base.py -------------------------------------------------------------------------------- /asdl/sql/parser/parser_v0.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/X-LANCE/text2sql-lgesql/HEAD/asdl/sql/parser/parser_v0.py -------------------------------------------------------------------------------- /asdl/sql/parser/parser_v1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/X-LANCE/text2sql-lgesql/HEAD/asdl/sql/parser/parser_v1.py -------------------------------------------------------------------------------- /asdl/sql/parser/parser_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/X-LANCE/text2sql-lgesql/HEAD/asdl/sql/parser/parser_v2.py -------------------------------------------------------------------------------- /asdl/sql/sql_transition_system.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/X-LANCE/text2sql-lgesql/HEAD/asdl/sql/sql_transition_system.py -------------------------------------------------------------------------------- /asdl/sql/unparser/unparser_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/X-LANCE/text2sql-lgesql/HEAD/asdl/sql/unparser/unparser_base.py -------------------------------------------------------------------------------- /asdl/sql/unparser/unparser_v0.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/X-LANCE/text2sql-lgesql/HEAD/asdl/sql/unparser/unparser_v0.py -------------------------------------------------------------------------------- /asdl/sql/unparser/unparser_v1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/X-LANCE/text2sql-lgesql/HEAD/asdl/sql/unparser/unparser_v1.py -------------------------------------------------------------------------------- /asdl/sql/unparser/unparser_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/X-LANCE/text2sql-lgesql/HEAD/asdl/sql/unparser/unparser_v2.py -------------------------------------------------------------------------------- /asdl/transition_system.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/X-LANCE/text2sql-lgesql/HEAD/asdl/transition_system.py -------------------------------------------------------------------------------- /eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/X-LANCE/text2sql-lgesql/HEAD/eval.py -------------------------------------------------------------------------------- /evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/X-LANCE/text2sql-lgesql/HEAD/evaluation.py -------------------------------------------------------------------------------- /model/decoder/onlstm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/X-LANCE/text2sql-lgesql/HEAD/model/decoder/onlstm.py -------------------------------------------------------------------------------- /model/decoder/sql_parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/X-LANCE/text2sql-lgesql/HEAD/model/decoder/sql_parser.py -------------------------------------------------------------------------------- /model/encoder/functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/X-LANCE/text2sql-lgesql/HEAD/model/encoder/functions.py -------------------------------------------------------------------------------- /model/encoder/graph_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/X-LANCE/text2sql-lgesql/HEAD/model/encoder/graph_encoder.py -------------------------------------------------------------------------------- /model/encoder/graph_input.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/X-LANCE/text2sql-lgesql/HEAD/model/encoder/graph_input.py -------------------------------------------------------------------------------- /model/encoder/graph_output.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/X-LANCE/text2sql-lgesql/HEAD/model/encoder/graph_output.py -------------------------------------------------------------------------------- /model/encoder/lgesql.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/X-LANCE/text2sql-lgesql/HEAD/model/encoder/lgesql.py -------------------------------------------------------------------------------- /model/encoder/rgatsql.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/X-LANCE/text2sql-lgesql/HEAD/model/encoder/rgatsql.py -------------------------------------------------------------------------------- /model/model_constructor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/X-LANCE/text2sql-lgesql/HEAD/model/model_constructor.py -------------------------------------------------------------------------------- /model/model_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/X-LANCE/text2sql-lgesql/HEAD/model/model_utils.py -------------------------------------------------------------------------------- /preprocess/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/X-LANCE/text2sql-lgesql/HEAD/preprocess/README.md -------------------------------------------------------------------------------- /preprocess/build_glove_vocab.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/X-LANCE/text2sql-lgesql/HEAD/preprocess/build_glove_vocab.py -------------------------------------------------------------------------------- /preprocess/common_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/X-LANCE/text2sql-lgesql/HEAD/preprocess/common_utils.py -------------------------------------------------------------------------------- /preprocess/get_tables.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/X-LANCE/text2sql-lgesql/HEAD/preprocess/get_tables.py -------------------------------------------------------------------------------- /preprocess/graph_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/X-LANCE/text2sql-lgesql/HEAD/preprocess/graph_utils.py -------------------------------------------------------------------------------- /preprocess/parse_raw_json.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/X-LANCE/text2sql-lgesql/HEAD/preprocess/parse_raw_json.py -------------------------------------------------------------------------------- /preprocess/parse_sql_one.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/X-LANCE/text2sql-lgesql/HEAD/preprocess/parse_sql_one.py -------------------------------------------------------------------------------- /preprocess/process_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/X-LANCE/text2sql-lgesql/HEAD/preprocess/process_dataset.py -------------------------------------------------------------------------------- /preprocess/process_graphs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/X-LANCE/text2sql-lgesql/HEAD/preprocess/process_graphs.py -------------------------------------------------------------------------------- /process_sql.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/X-LANCE/text2sql-lgesql/HEAD/process_sql.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/X-LANCE/text2sql-lgesql/HEAD/requirements.txt -------------------------------------------------------------------------------- /results/results_bert.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/X-LANCE/text2sql-lgesql/HEAD/results/results_bert.txt -------------------------------------------------------------------------------- /results/results_electra.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/X-LANCE/text2sql-lgesql/HEAD/results/results_electra.txt -------------------------------------------------------------------------------- /results/results_glove.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/X-LANCE/text2sql-lgesql/HEAD/results/results_glove.txt -------------------------------------------------------------------------------- /run/run_evaluation.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/X-LANCE/text2sql-lgesql/HEAD/run/run_evaluation.sh -------------------------------------------------------------------------------- /run/run_lgesql_glove.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/X-LANCE/text2sql-lgesql/HEAD/run/run_lgesql_glove.sh -------------------------------------------------------------------------------- /run/run_lgesql_plm.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/X-LANCE/text2sql-lgesql/HEAD/run/run_lgesql_plm.sh -------------------------------------------------------------------------------- /run/run_preprocessing.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/X-LANCE/text2sql-lgesql/HEAD/run/run_preprocessing.sh -------------------------------------------------------------------------------- /run/run_submission.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/X-LANCE/text2sql-lgesql/HEAD/run/run_submission.sh -------------------------------------------------------------------------------- /scripts/text2sql.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/X-LANCE/text2sql-lgesql/HEAD/scripts/text2sql.py -------------------------------------------------------------------------------- /setup.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/X-LANCE/text2sql-lgesql/HEAD/setup.sh -------------------------------------------------------------------------------- /utils/args.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/X-LANCE/text2sql-lgesql/HEAD/utils/args.py -------------------------------------------------------------------------------- /utils/batch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/X-LANCE/text2sql-lgesql/HEAD/utils/batch.py -------------------------------------------------------------------------------- /utils/constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/X-LANCE/text2sql-lgesql/HEAD/utils/constants.py -------------------------------------------------------------------------------- /utils/evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/X-LANCE/text2sql-lgesql/HEAD/utils/evaluator.py -------------------------------------------------------------------------------- /utils/example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/X-LANCE/text2sql-lgesql/HEAD/utils/example.py -------------------------------------------------------------------------------- /utils/graph_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/X-LANCE/text2sql-lgesql/HEAD/utils/graph_example.py -------------------------------------------------------------------------------- /utils/hyperparams.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/X-LANCE/text2sql-lgesql/HEAD/utils/hyperparams.py -------------------------------------------------------------------------------- /utils/initialization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/X-LANCE/text2sql-lgesql/HEAD/utils/initialization.py -------------------------------------------------------------------------------- /utils/optimization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/X-LANCE/text2sql-lgesql/HEAD/utils/optimization.py -------------------------------------------------------------------------------- /utils/vocab.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/X-LANCE/text2sql-lgesql/HEAD/utils/vocab.py -------------------------------------------------------------------------------- /utils/word2vec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/X-LANCE/text2sql-lgesql/HEAD/utils/word2vec.py --------------------------------------------------------------------------------