mathemakitten commited on
Commit
72b6360
1 Parent(s): 555a15f
Files changed (1) hide show
  1. glue-suite-v2.py +3 -2
glue-suite-v2.py CHANGED
@@ -1,5 +1,6 @@
1
  from datasets.features import Features, Sequence, Value
2
  from evaluate.module import EvaluationModuleInfo
 
3
  import evaluate
4
 
5
 
@@ -12,7 +13,7 @@ class Suite(evaluate.EvaluationSuite):
12
  features=Features({"predictions": Value("int64"), "references": Value("int64")}))
13
  preprocessor = None #lambda x: x["text"].lower()
14
  suite = [
15
- evaluate.evaluation_suite.SubTask(
16
  data="imdb",
17
  split="test",
18
  data_preprocessor=preprocessor,
@@ -26,7 +27,7 @@ class Suite(evaluate.EvaluationSuite):
26
  }
27
  }
28
  ),
29
- evaluate.evaluation_suite.SubTask(
30
  data="sst2",
31
  split="test[:10]",
32
  data_preprocessor=preprocessor,
 
1
  from datasets.features import Features, Sequence, Value
2
  from evaluate.module import EvaluationModuleInfo
3
+ from evaluate.evaluation_suite import SubTask
4
  import evaluate
5
 
6
 
 
13
  features=Features({"predictions": Value("int64"), "references": Value("int64")}))
14
  preprocessor = None #lambda x: x["text"].lower()
15
  suite = [
16
+ SubTask(
17
  data="imdb",
18
  split="test",
19
  data_preprocessor=preprocessor,
 
27
  }
28
  }
29
  ),
30
+ SubTask(
31
  data="sst2",
32
  split="test[:10]",
33
  data_preprocessor=preprocessor,