t5-prompt.yaml 514 Bytes
Newer Older
lintangsutawika's avatar
lintangsutawika committed
1
2
group:
  - super-glue-t5-prompt
3
task: super_glue-wsc-t5-prompt
lintangsutawika's avatar
lintangsutawika committed
4
dataset_path: super_glue
lintangsutawika's avatar
lintangsutawika committed
5
dataset_name: wsc.fixed
lintangsutawika's avatar
lintangsutawika committed
6
7
training_split: train
validation_split: validation
8
output_type: generate_until
lintangsutawika's avatar
lintangsutawika committed
9
10
doc_to_text: !function "t5_utils.doc_to_text"
doc_to_target: label
lintangsutawika's avatar
lintangsutawika committed
11
12
13
generation_kwargs:
  until:
    - "</s>"
lintangsutawika's avatar
lintangsutawika committed
14
metric_list:
lintangsutawika's avatar
lintangsutawika committed
15
  - metric: accuracy
lintangsutawika's avatar
lintangsutawika committed
16
17
    aggregation: mean
    higher_is_better: true
lintangsutawika's avatar
lintangsutawika committed
18
19
20
filter_list:
  - name: "wsc_postprocessor"
    filter:
lintangsutawika's avatar
lintangsutawika committed
21
      - function: !function t5_utils.WSCPostprocess
lintangsutawika's avatar
lintangsutawika committed
22
metadata:
23
  version: 0.0