Change DPR accuracy reporting to 4 digits: better implementation (#1094)

Ref #1092
castorini · Mar 27, 2022 · 5668edd · 5668edd
1 parent 8d2a67c
commit 5668edd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyserini/eval/evaluate_dpr_retrieval.py b/pyserini/eval/evaluate_dpr_retrieval.py
@@ -266,7 +266,7 @@ def evaluate_retrieval(retrieval_file, topk, regex=False):
             accuracy[k].append(0 if has_ans_idx >= k else 1)
 
     for k in topk:
-        print(f'Top{k}\taccuracy: {np.around(np.mean(accuracy[k]), 4)}')
+        print(f'Top{k}\taccuracy: {np.mean(accuracy[k]):.4f}')
 
 
 if __name__ == '__main__':