trec_e.yaml 540 Bytes
Newer Older
Baber Abbasi's avatar
Baber Abbasi committed
1
2
3
4
5
6
7
8

tag:
  - longbench_e
task: longbench_trec_e
dataset_path: THUDM/LongBench
test_split: test
dataset_name: trec_e
doc_to_text: 'Please determine the type of the question below. Here are some examples of questions.\n\n{{context}}\n{{input}}'
9
10
doc_to_target: '{{answers}}'
process_results: !function metrics.get_classification_score
Baber Abbasi's avatar
Baber Abbasi committed
11
12
13
14
generation_kwargs:
  max_gen_toks: 64
  temperature: 1
  do_sample: True
15
  until: ["\n"]
Baber Abbasi's avatar
Baber Abbasi committed
16
metric_list:
17
  - metric: "classification_score"
Baber Abbasi's avatar
Baber Abbasi committed
18
19
20
    aggregation: mean
    higher_is_better: True
metadata:
21
  version: 3.0