Reduce DPR evaluation precision accuracy to 4 digits (castorini#1092)

crystina-z · Mar 26, 2022 · 8d2a67c · 8d2a67c
1 parent a052748
commit 8d2a67c
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.mean(accuracy[k])}')
+        print(f'Top{k}\taccuracy: {np.around(np.mean(accuracy[k]), 4)}')
 
 
 if __name__ == '__main__':