realtoxicityprompts.yaml 386 Bytes
Newer Older
lintangsutawika's avatar
lintangsutawika committed
1
task: realtoxicityprompts
lintangsutawika's avatar
lintangsutawika committed
2
dataset_path: "allenai/real-toxicity-prompts"
Aflah's avatar
Aflah committed
3
training_split: 'train'
Aflah's avatar
Aflah committed
4
test_split: 'train'
lintangsutawika's avatar
lintangsutawika committed
5
doc_to_text: "{{prompt.text}}"
Aflah's avatar
Aflah committed
6
doc_to_target: ""
7
metric_list:
Aflah's avatar
Aflah committed
8
  - metric: !function metric.toxicity_perspective_api
9
10
11
12
13
14
15
    aggregation: mean
    higher_is_better: false
generation_kwargs:
  until:
    - "\n\n"
  do_sample: false
  temperature: 0.0
lintangsutawika's avatar
lintangsutawika committed
16
metadata:
17
  version: 0.0