forked from wbbeyourself/MAC-SQL
-
Notifications
You must be signed in to change notification settings - Fork 0
/
evaluation_bird_ex_ves.sh
36 lines (33 loc) · 1.12 KB
/
evaluation_bird_ex_ves.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
#!/bin/bash
db_root_path="./data/bird/dev_databases/"
data_mode="dev"
diff_json_path="./data/bird/dev.json"
predicted_sql_json_path="./outputs/bird/predict_dev.json"
ground_truth_sql_path="./data/bird/dev_gold.sql"
num_cpus=12
meta_time_out=30.0
time_out=60
mode_gt="gt"
mode_predict="gpt"
# evaluate EX
echo "Evaluate BIRD EX begin!"
python ./evaluation/evaluation_bird_ex.py --db_root_path $db_root_path \
--predicted_sql_json_path $predicted_sql_json_path \
--data_mode $data_mode \
--ground_truth_sql_path $ground_truth_sql_path \
--num_cpus $num_cpus \
--mode_predict $mode_predict \
--diff_json_path $diff_json_path \
--meta_time_out $meta_time_out
echo "Evaluate EX done!"
# evaluate VES
echo "Evaluate BIRD VES begin!"
python ./evaluation/evaluation_bird_ves.py \
--db_root_path $db_root_path \
--predicted_sql_json_path $predicted_sql_json_path \
--data_mode $data_mode \
--ground_truth_sql_path $ground_truth_sql_path \
--num_cpus $num_cpus --meta_time_out $time_out \
--mode_gt $mode_gt --mode_predict $mode_predict \
--diff_json_path $diff_json_path
echo "Evaluate VES done!"