yujunjun/haystack
1
0
Fork 0
mirror of https://github.com/deepset-ai/haystack.git synced 2025-11-14 09:03:34 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix offsets in reader (#4)

Browse Source
This commit is contained in:
Malte Pietsch 2020-01-14 17:28:29 +01:00 committed by GitHub
parent aea954c03a
commit 0b3fd7572b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
haystack/reader/farm.py
View File

@ -85,10 +85,10 @@ class FARMReader:
if a["answer"]: #skip "no answer" if a["answer"]: #skip "no answer"
cur = {"answer": a["answer"], cur = {"answer": a["answer"],
"score": a["score"], "score": a["score"],
"probability": expit(np.asarray([a["score"]]) / 8), #just a pseudo prob for now "probability": float(expit(np.asarray([a["score"]]) / 8)), #just a pseudo prob for now
"context": a["context"], "context": a["context"],
"offset_start": a["offset_answer_start"] - a["offset_context_start"], "offset_start": a["offset_answer_start"] - a["offset_context_start"],
"offset_end": a["offset_answer_start"] - a["offset_context_start"], "offset_end": a["offset_answer_end"] - a["offset_context_start"],
"document_id": a["document_id"]} "document_id": a["document_id"]}
answers.append(cur) answers.append(cur)