topdog/supersonic
1
0
Fork 0
mirror of https://github.com/tencentmusic/supersonic.git synced 2026-04-18 20:34:19 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

(improvement)(Headless) corrector supports subselect sql (#1006)

Browse Source 
* [improvement] corrector support subselect sql

---------

Co-authored-by: zuopengge
This commit is contained in:
mainmain
2024-05-17 10:01:49 +08:00
committed by GitHub
parent 2411cb3fdc
commit 7949efea75
6 changed files with 32 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
evaluation/build_models.py
View File

@@ -258,7 +258,7 @@ def build_dataSet(domain_id,model_id1,model_id2,model_id3,model_id4):
{"id":model_id3,"includesAll":False,"metrics":metric_list3,"dimensions":dimension_list3},
{"id":model_id4,"includesAll":False,"metrics":metric_list4,"dimensions":dimension_list4}
]},
"queryConfig":{"tagTypeDefaultConfig":{},"metricTypeDefaultConfig":{"timeDefaultConfig":{"unit":1,"period":"DAY","timeMode":"RECENT"}}},"admins":["admin"],"admin":"admin"}
"queryConfig":{"tagTypeDefaultConfig":{},"metricTypeDefaultConfig":{"timeDefaultConfig":{"unit":0,"period":"DAY","timeMode":"RECENT"}}},"admins":["admin"],"admin":"admin"}
url=get_url_pre()+"/api/semantic/dataSet"
authorization=get_authorization()
header = {}

3
evaluation/build_pred_result.py
View File

@@ -66,6 +66,7 @@ def get_pred_result():
pred_sql_list=[]
default_sql="select * from tablea "
time_cost=[]
time.sleep(60)
for i in range(0,len(questions)):
start_time = time.time()
pred_sql=get_pred_sql(questions[i],url,agent_id,chat_id,authorization,default_sql)
@@ -73,7 +74,7 @@ def get_pred_result():
cost='%.3f'%(end_time-start_time)
time_cost.append(cost)
pred_sql_list.append(pred_sql)
time.sleep(60)
time.sleep(3)
write_sql(pred_sql_path, pred_sql_list)
return [float(cost) for cost in time_cost]