Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -489,8 +489,8 @@ def run_evaluation(file_obj, scenario):
|
|
| 489 |
rel_dict = ground_truth.get(q_id, {})
|
| 490 |
relevant_kodes_set = {k for k, r in rel_dict.items() if r > 0}
|
| 491 |
|
| 492 |
-
mrr = calculate_mrr(retrieved_kodes,
|
| 493 |
-
recall = calculate_recall(retrieved_kodes,
|
| 494 |
ndcg = calculate_ndcg(retrieved_kodes, rel_dict, k=10)
|
| 495 |
|
| 496 |
results_list.append({
|
|
|
|
| 489 |
rel_dict = ground_truth.get(q_id, {})
|
| 490 |
relevant_kodes_set = {k for k, r in rel_dict.items() if r > 0}
|
| 491 |
|
| 492 |
+
mrr = calculate_mrr(retrieved_kodes, relevant_kodes_set)
|
| 493 |
+
recall = calculate_recall(retrieved_kodes, relevant_kodes_set, k=10)
|
| 494 |
ndcg = calculate_ndcg(retrieved_kodes, rel_dict, k=10)
|
| 495 |
|
| 496 |
results_list.append({
|