allbibek commited on
Commit
95542db
·
verified ·
1 Parent(s): f54c21c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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, relevant_kodes)
493
- recall = calculate_recall(retrieved_kodes, relevant_kodes, k=10)
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({