Change to retriever eval top_k to match notebook

This commit is contained in:
brandenchan 2020-08-18 11:39:49 +02:00
parent b30963d0cd
commit 8a3eca05c3

View File

@ -80,7 +80,7 @@ finder = Finder(reader, retriever)
## Evaluate Retriever on its own
if eval_retriever_only:
retriever_eval_results = retriever.eval(top_k=1, label_index=label_index, doc_index=doc_index)
retriever_eval_results = retriever.eval(top_k=10, label_index=label_index, doc_index=doc_index)
## Retriever Recall is the proportion of questions for which the correct document containing the answer is
## among the correct documents
print("Retriever Recall:", retriever_eval_results["recall"])