Update test_rag.py
Browse files- 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 |
# ------------------------------
|