diff --git a/tutorials/Tutorial5_Evaluation.ipynb b/tutorials/Tutorial5_Evaluation.ipynb index 168109518..db06dca8d 100644 --- a/tutorials/Tutorial5_Evaluation.ipynb +++ b/tutorials/Tutorial5_Evaluation.ipynb @@ -297,7 +297,7 @@ "# Initialize Reader\n", "from haystack.reader.farm import FARMReader\n", "\n", - "reader = FARMReader(\"deepset/roberta-base-squad2\")" + "reader = FARMReader(\"deepset/roberta-base-squad2\", top_k_per_candidate=4)" ] }, { @@ -1957,4 +1957,4 @@ }, "nbformat": 4, "nbformat_minor": 1 -} +} \ No newline at end of file diff --git a/tutorials/Tutorial5_Evaluation.py b/tutorials/Tutorial5_Evaluation.py index 7c64b4ce5..78b78369d 100644 --- a/tutorials/Tutorial5_Evaluation.py +++ b/tutorials/Tutorial5_Evaluation.py @@ -75,7 +75,7 @@ retriever = ElasticsearchRetriever(document_store=document_store) # Initialize Reader -reader = FARMReader("deepset/roberta-base-squad2") +reader = FARMReader("deepset/roberta-base-squad2", top_k_per_candidate=4) # Initialize Finder which sticks together Reader and Retriever finder = Finder(reader, retriever)