mathemakitten commited on
Commit
26647e7
·
1 Parent(s): aac9ec9
Files changed (1) hide show
  1. glue-suite-v2.py +9 -9
glue-suite-v2.py CHANGED
@@ -19,15 +19,15 @@ class Suite(evaluate.EvaluationSuite):
19
  data="imdb",
20
  split="test",
21
  data_preprocessor=self.preprocessor,
22
- # args_for_task={
23
- # "metric": "accuracy",
24
- # "input_column": "text",
25
- # "label_column": "label",
26
- # "label_mapping": {
27
- # "LABEL_0": 0.0,
28
- # "LABEL_1": 1.0
29
- # }
30
- # }
31
  ),
32
  SubTask(
33
  data="sst2",
 
19
  data="imdb",
20
  split="test",
21
  data_preprocessor=self.preprocessor,
22
+ args_for_task={
23
+ "metric": "accuracy",
24
+ "input_column": "text",
25
+ "label_column": "label",
26
+ "label_mapping": {
27
+ "LABEL_0": 0.0,
28
+ "LABEL_1": 1.0
29
+ }
30
+ }
31
  ),
32
  SubTask(
33
  data="sst2",