t5-prompt.yaml 461 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
doc_to_text: !function "t5_utils.doc_to_text"
10
process_results: !function "t5_utils.process_results"
lintangsutawika's avatar
lintangsutawika committed
11
doc_to_target: label
lintangsutawika's avatar
lintangsutawika committed
12
13
14
generation_kwargs:
  until:
    - "</s>"
lintangsutawika's avatar
lintangsutawika committed
15
metric_list:
lintangsutawika's avatar
lintangsutawika committed
16
  - metric: accuracy
lintangsutawika's avatar
lintangsutawika committed
17
18
    aggregation: mean
    higher_is_better: true
lintangsutawika's avatar
lintangsutawika committed
19
metadata:
20
  version: 1.0