topdog/supersonic
1
0
Fork 0
mirror of https://github.com/tencentmusic/supersonic.git synced 2025-12-11 03:58:14 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

(improvement) improve evalution accuracy (#727)

Browse Source
This commit is contained in:
mainmain
2024-02-19 17:31:38 +08:00
committed by GitHub
parent fdb69547e6
commit 699a33b1c1
13 changed files with 112 additions and 53 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
evaluation/build_pred_result.py
View File

@@ -25,7 +25,7 @@ def get_pred_sql(query,url,agentId,chatId,authorization,default_sql):
header["Authorization"] =authorization
try:
result = requests.post(url=url, headers=header, json=data)
print(result.json())
#print(result.json())
print(result.json()["traceId"])
if result.status_code == 200:
data = result.json()["data"]
@@ -68,7 +68,7 @@ def get_pred_result():
for i in range(0,len(questions)):
pred_sql=get_pred_sql(questions[i],url,agent_id,chat_id,authorization,default_sql)
pred_sql_list.append(pred_sql)
time.sleep(30)
time.sleep(60)
write_sql(pred_sql_path, pred_sql_list)
if __name__ == "__main__":