yujunjun/haystack
1
0
Fork 0
mirror of https://github.com/deepset-ai/haystack.git synced 2025-08-31 11:56:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add document meta for Transformer Reader (#114)

Browse Source
This commit is contained in:
Stan Kirdey 2020-05-18 06:31:54 -07:00 committed by GitHub
parent bf8e506c45
commit f4455ee42f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
haystack/reader/transformers.py
View File

@ -87,7 +87,8 @@ class TransformersReader:
"offset_answer_end": pred["end"], "offset_answer_end": pred["end"],
"probability": pred["score"], "probability": pred["score"],
"score": None, "score": None,
"document_id": doc.id "document_id": doc.id,
"meta": doc.meta
}) })
# sort answers by their `probability` and select top-k # sort answers by their `probability` and select top-k