Add metadata for FAQ style QA (#106)

This commit is contained in:
Tanay Soni 2020-05-12 15:41:46 +02:00 committed by GitHub
parent 42b60b5578
commit 73aed42f14
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -86,7 +86,7 @@ class Finder:
for doc in documents:
#TODO proper calibratation of pseudo probabilities
cur_answer = {"question": doc.meta["question"], "answer": doc.text, "context": doc.text,
"score": doc.query_score, "offset_start": 0, "offset_end": len(doc.text),
"score": doc.query_score, "offset_start": 0, "offset_end": len(doc.text), "meta": doc.meta
}
if self.retriever.embedding_model:
probability = (doc.query_score + 1) / 2