Commit 6469a0e7 authored by Leo Gao's avatar Leo Gao
Browse files

Replace HFNLPTask with HFTask everywhere

parent f348fa2c
import base
class CoQA(base.Dataset):
def training_docs(self):
pass
def validation_docs(self):
pass
def test_docs(self):
pass
def fewshot_description(self):
pass
\ No newline at end of file
from . common import HFNLPTask from . common import HFTask
class ARCEasy(HFNLPTask): class ARCEasy(HFTask):
NLP_PATH = "ai2_arc" NLP_PATH = "ai2_arc"
NLP_NAME = "ARC-Easy" NLP_NAME = "ARC-Easy"
......
...@@ -3,7 +3,7 @@ import json ...@@ -3,7 +3,7 @@ import json
from scipy.stats import pearsonr, spearmanr from scipy.stats import pearsonr, spearmanr
from sklearn.metrics import f1_score, matthews_corrcoef from sklearn.metrics import f1_score, matthews_corrcoef
from tqdm import auto as tqdm_lib from tqdm import auto as tqdm_lib
from . common import HFNLPTask, simple_accuracy_metric, yesno from . common import HFTask, simple_accuracy_metric, yesno
from pathlib import Path from pathlib import Path
from ..base import Dataset from ..base import Dataset
......
from . common import HFNLPTask from . common import HFTask
from ..utils_stream import X, each, apply, join, filt, one from ..utils_stream import X, each, apply, join, filt, one
import collections import collections
import nlp import nlp
class RACE(HFNLPTask): class RACE(HFTask):
NLP_PATH = "race" NLP_PATH = "race"
NLP_NAME = "high" NLP_NAME = "high"
......
import numpy as np import numpy as np
from tqdm import auto as tqdm_lib from tqdm import auto as tqdm_lib
from . common import HFNLPTask, simple_accuracy_metric, yesno from . common import HFTask, simple_accuracy_metric, yesno
class BoolQ(HFNLPTask): class BoolQ(HFTask):
NLP_PATH = "super_glue" NLP_PATH = "super_glue"
NLP_NAME = "boolq" NLP_NAME = "boolq"
...@@ -36,7 +36,7 @@ class BoolQ(HFNLPTask): ...@@ -36,7 +36,7 @@ class BoolQ(HFNLPTask):
return simple_accuracy_metric(preds=preds, golds=golds) return simple_accuracy_metric(preds=preds, golds=golds)
class CommitmentBank(HFNLPTask): class CommitmentBank(HFTask):
NLP_PATH = "super_glue" NLP_PATH = "super_glue"
NLP_NAME = "cb" NLP_NAME = "cb"
...@@ -79,7 +79,7 @@ class CommitmentBank(HFNLPTask): ...@@ -79,7 +79,7 @@ class CommitmentBank(HFNLPTask):
return simple_accuracy_metric(preds=preds, golds=golds) return simple_accuracy_metric(preds=preds, golds=golds)
class Copa(HFNLPTask): class Copa(HFTask):
NLP_PATH = "super_glue" NLP_PATH = "super_glue"
NLP_NAME = "copa" NLP_NAME = "copa"
...@@ -120,7 +120,7 @@ class Copa(HFNLPTask): ...@@ -120,7 +120,7 @@ class Copa(HFNLPTask):
return choice[0].lower() + choice[1:] return choice[0].lower() + choice[1:]
class MultiRC(HFNLPTask): class MultiRC(HFTask):
NLP_PATH = "super_glue" NLP_PATH = "super_glue"
NLP_NAME = "multirc" NLP_NAME = "multirc"
...@@ -177,7 +177,7 @@ class MultiRC(HFNLPTask): ...@@ -177,7 +177,7 @@ class MultiRC(HFNLPTask):
} }
class WordsInContext(HFNLPTask): class WordsInContext(HFTask):
NLP_PATH = "super_glue" NLP_PATH = "super_glue"
NLP_NAME = "wic" NLP_NAME = "wic"
...@@ -214,7 +214,7 @@ class WordsInContext(HFNLPTask): ...@@ -214,7 +214,7 @@ class WordsInContext(HFNLPTask):
return simple_accuracy_metric(preds=preds, golds=golds) return simple_accuracy_metric(preds=preds, golds=golds)
class WinogradSchemaChallenge(HFNLPTask): class WinogradSchemaChallenge(HFTask):
NLP_PATH = "super_glue" NLP_PATH = "super_glue"
NLP_NAME = "wsc" NLP_NAME = "wsc"
......
from . common import HFNLPTask from . common import HFTask
class WebQs(HFNLPTask): class WebQs(HFTask):
NLP_PATH = "web_questions" NLP_PATH = "web_questions"
NLP_NAME = None NLP_NAME = None
......
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