Commit 155aeee9 authored by Jon Tow's avatar Jon Tow
Browse files

Rename list of loglikelihoods for consistency

parent 43e274f8
......@@ -59,12 +59,12 @@ class Winogrande(HFTask):
part of the document for `doc`.
"""
target = self.partial_target(doc)
ll = []
lls = []
for option in [doc["option1"], doc["option2"]]:
partial_ctx = self.partial_context(doc, option)
full_ctx = self.append_context(ctx, partial_ctx)
ll.append(rf.loglikelihood(full_ctx, target)[0])
return ll
lls.append(rf.loglikelihood(full_ctx, target)[0])
return lls
@classmethod
def append_context(cls, ctx, partial_ctx):
......
......@@ -90,12 +90,12 @@ class WinogradSchemaChallenge273(HFTask):
part of the document for `doc`.
"""
target = self.partial_target(doc)
ll = []
lls = []
for option in doc["options"]:
partial_ctx = self.partial_context(doc, option)
full_ctx = self.append_context(ctx, partial_ctx)
ll.append(rf.loglikelihood(full_ctx, target)[0])
return ll
lls.append(rf.loglikelihood(full_ctx, target)[0])
return lls
@classmethod
def append_context(cls, ctx, partial_ctx):
......
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