yujunjun/haystack
1
0
Fork 0
mirror of https://github.com/deepset-ai/haystack.git synced 2025-08-28 02:16:32 +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",
"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
}
}

2
tutorials/Tutorial5_Evaluation.py
View File

@ -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)