yujunjun/haystack
1
0
Fork 0
mirror of https://github.com/deepset-ai/haystack.git synced 2025-08-28 10:26:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

Set top_k_per_candidate

Browse Source
This commit is contained in:
brandenchan 2020-08-26 12:03:56 +02:00
parent cca8676f90
commit b44b1ac6ec
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tutorials/Tutorial5_Evaluation.ipynb
View File

@ -297,7 +297,7 @@
"# Initialize Reader\n", "# Initialize Reader\n",
"from haystack.reader.farm import FARMReader\n", "from haystack.reader.farm import FARMReader\n",
"\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": 4,
"nbformat_minor": 1 "nbformat_minor": 1
} }

2
tutorials/Tutorial5_Evaluation.py
View File

@ -75,7 +75,7 @@ retriever = ElasticsearchRetriever(document_store=document_store)
# Initialize Reader # 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 # Initialize Finder which sticks together Reader and Retriever
finder = Finder(reader, retriever) finder = Finder(reader, retriever)