orionweller commited on
Commit
05ff7af
·
1 Parent(s): 99526c6
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -253,7 +253,7 @@ def run_evaluation(dataset, postfix):
253
  logging.info(f"Number of queries in q_lookups: {len(q_lookups[dataset])}")
254
  logging.info("Size of all_scores: " + str(len(all_scores)))
255
  logging.info("Size of psg_indices: " + str(len(psg_indices)))
256
- for query, scores, doc_ids in zip(queries, all_scores, psg_indices):
257
  qid = query2qid[dataset][query]
258
  qid_str = str(qid)
259
  results[qid_str] = {}
 
253
  logging.info(f"Number of queries in q_lookups: {len(q_lookups[dataset])}")
254
  logging.info("Size of all_scores: " + str(len(all_scores)))
255
  logging.info("Size of psg_indices: " + str(len(psg_indices)))
256
+ for query, scores, doc_ids in zip(queries[current_dataset], all_scores, psg_indices):
257
  qid = query2qid[dataset][query]
258
  qid_str = str(qid)
259
  results[qid_str] = {}