"test/vscode:/vscode.git/clone" did not exist on "5913609168e6f691185f1b7b214e517d67dd5075"
Commit 26e94211 authored by cjlovering's avatar cjlovering
Browse files

MRPC with PS integration.

parent b44aa554
......@@ -222,7 +222,7 @@ class RTE(PromptSourceTask):
# Similarity and Paraphrase Tasks
class MRPC(Task):
class MRPC(PromptSourceTask):
VERSION = 0
DATASET_PATH = "glue"
DATASET_NAME = "mrpc"
......@@ -244,21 +244,6 @@ class MRPC(Task):
def validation_docs(self):
return self.dataset["validation"]
def process_results(self, doc, results):
ll_yes, ll_no = results
gold = doc["label"]
pred = ll_yes > ll_no
return {
"acc": pred == gold,
"f1": (gold, pred),
}
def higher_is_better(self):
return {"acc": True, "f1": True}
def aggregation(self):
return {"acc": mean, "f1": f1_score}
class QQP(Task):
VERSION = 0
......
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