Set top_k_per_candidate

deepset-ai · Aug 26, 2020 · b44b1ac · b44b1ac
1 parent cca8676
commit b44b1ac
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git 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
-}
+}
diff --git a/tutorials/Tutorial5_Evaluation.py b/tutorials/Tutorial5_Evaluation.py
@@ -75,7 +75,7 @@
 
 
 # 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)