Commit 26e55d33 authored by ingyuseong's avatar ingyuseong
Browse files

Fix KLUE few-shot label

parent 94c2c064
......@@ -58,11 +58,11 @@ class STS(Task):
)
def doc_to_target(self, doc):
return " {}".format({1: "", 0: " 아니"}[doc["labels"]["binary-label"]])
return " {}".format({0: "아니오", 1: ""}[doc["labels"]["binary-label"]])
def construct_requests(self, doc, ctx):
ll_positive, _ = rf.loglikelihood(ctx, " 예")
ll_negative, _ = rf.loglikelihood(ctx, " 아니")
ll_negative, _ = rf.loglikelihood(ctx, " 아니")
return ll_positive, ll_negative
def process_results(self, doc, results):
......
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