scroll_summary_task_yaml 670 Bytes
Newer Older
lintangsutawika's avatar
lintangsutawika committed
1
2
3
4
5
group: scrolls
dataset_path: tau/scrolls
output_type: greedy_until
training_split: train
validation_split: validation
lintangsutawika's avatar
update  
lintangsutawika committed
6
process_docs: !function preprocessors.process_docs
lintangsutawika's avatar
lintangsutawika committed
7
doc_to_text: "{{input}}\n\nQuestion: What is a summary of the preceding text?\nAnswer:"
lintangsutawika's avatar
update  
lintangsutawika committed
8
doc_to_target: "{{outputs|join(', ')}}"
lintangsutawika's avatar
lintangsutawika committed
9
10
11
should_decontaminate: true
doc_to_decontamination_query: input
metric_list:
lintangsutawika's avatar
update  
lintangsutawika committed
12
13
14
15
16
17
18
19
20
  - metric: !function metrics.rouge1
    aggregation: mean
    higher_is_better: true
  - metric: !function metrics.rouge2
    aggregation: mean
    higher_is_better: true
  - metric: !function metrics.rougeL
    aggregation: mean
    higher_is_better: true
lintangsutawika's avatar
lintangsutawika committed
21
22
23
generation_kwargs:
  until:
    - "\n"