|
set -e |
|
|
|
|
|
python text2sql_data_generator.py \ |
|
--input_dataset_path "./data/preprocessed_data/preprocessed_train_spider_amr.json" \ |
|
--output_dataset_path "./data/preprocessed_data/resdsql_train_spider_amr.json" \ |
|
--topk_table_num 4 \ |
|
--topk_column_num 5 \ |
|
--mode "train" \ |
|
--noise_rate 0.2 \ |
|
--use_contents \ |
|
--add_fk_info \ |
|
--output_skeleton \ |
|
--target_type "sql" |
|
|
|
|
|
python schema_item_classifier.py \ |
|
--batch_size 32 \ |
|
--device "0" \ |
|
--seed 42 \ |
|
--save_path "./models/text2sql_schema_item_classifier_semantic" \ |
|
--dev_filepath "./data/preprocessed_data/preprocessed_dev_amr.json" \ |
|
--output_filepath "./data/preprocessed_data/dev_with_probs_amr.json" \ |
|
--use_contents \ |
|
--add_fk_info \ |
|
--mode "eval" |
|
|
|
|
|
python text2sql_data_generator.py \ |
|
--input_dataset_path "./data/preprocessed_data/dev_with_probs_amr.json" \ |
|
--output_dataset_path "./data/preprocessed_data/resdsql_dev_amr.json" \ |
|
--topk_table_num 4 \ |
|
--topk_column_num 5 \ |
|
--mode "eval" \ |
|
--use_contents \ |
|
--add_fk_info \ |
|
--output_skeleton \ |
|
--target_type "sql" |
|
|