MikeTerekhov commited on
Commit
ace09b0
·
verified ·
1 Parent(s): 5330d88

Update test_rag.py

Browse files
Files changed (1) hide show
  1. test_rag.py +5 -0
test_rag.py CHANGED
@@ -355,6 +355,11 @@ Request: {row["natural_query"]}
355
  print("Percent SQLite matched: " + str(num_sql_matched / len(nba_df)))
356
  print("Percent result matched: " + str(num_result_matched / len(nba_df)))
357
  print("Dataset length: " + str(len(nba_df)))
 
 
 
 
 
358
 
359
 
360
  # ------------------------------
 
355
  print("Percent SQLite matched: " + str(num_sql_matched / len(nba_df)))
356
  print("Percent result matched: " + str(num_result_matched / len(nba_df)))
357
  print("Dataset length: " + str(len(nba_df)))
358
+ print("-------------------")
359
+ print("Num queries tested: ", counter)
360
+ print("Num correct queries: ", num_result_matched)
361
+ print("Acc: ", (num_result_matched / counter)*100)
362
+ print("-------------------")
363
 
364
 
365
  # ------------------------------