samsum.yaml 522 Bytes
Newer Older
Baber Abbasi's avatar
Baber Abbasi committed
1
2
3
4
5
6
7

tag:
  - longbench
task: longbench_samsum
dataset_path: THUDM/LongBench
test_split: test
dataset_name: samsum
Timur Aysin's avatar
Timur Aysin committed
8
doc_to_text: "Summarize the dialogue into a few short sentences. The following are some examples.\n\n{{context}}\n\n{{input}}"
9
10
doc_to_target: '{{answers}}'
process_results: !function metrics.get_rouge_score
Baber Abbasi's avatar
Baber Abbasi committed
11
12
13
generation_kwargs:
  max_gen_toks: 128
  temperature: 1
Timur Aysin's avatar
Timur Aysin committed
14
  do_sample: False
15
  until: ["\n"]
Baber Abbasi's avatar
Baber Abbasi committed
16
metric_list:
17
  - metric: "rouge_score"
Baber Abbasi's avatar
Baber Abbasi committed
18
19
20
    aggregation: mean
    higher_is_better: True
metadata:
Timur Aysin's avatar
Timur Aysin committed
21
  version: 4.0