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

MRPC with PS integration.

parent b44aa554
...@@ -222,7 +222,7 @@ class RTE(PromptSourceTask): ...@@ -222,7 +222,7 @@ class RTE(PromptSourceTask):
# Similarity and Paraphrase Tasks # Similarity and Paraphrase Tasks
class MRPC(Task): class MRPC(PromptSourceTask):
VERSION = 0 VERSION = 0
DATASET_PATH = "glue" DATASET_PATH = "glue"
DATASET_NAME = "mrpc" DATASET_NAME = "mrpc"
...@@ -244,21 +244,6 @@ class MRPC(Task): ...@@ -244,21 +244,6 @@ class MRPC(Task):
def validation_docs(self): def validation_docs(self):
return self.dataset["validation"] 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): class QQP(Task):
VERSION = 0 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