Commit 183f4721 authored by Kevin Wang's avatar Kevin Wang
Browse files

prepending a space, because it significantly increases log-likelihood of the correct answer.

parent a58e5e4b
......@@ -40,7 +40,7 @@ class Arithmetic(Dataset):
return doc.answer_text
def construct_requests(self, doc, ctx):
ll, is_prediction = rf.loglikelihood(ctx, doc.answer_text)
ll, is_prediction = rf.loglikelihood(ctx, ' '+doc.answer_text)
# not sure what the difference between the two objects returned by rf.loglikehood are here
return is_prediction
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment