Commit 1a77b4d5 authored by haileyschoelkopf's avatar haileyschoelkopf
Browse files

update generate_tasks.py and template yamls

parent 605787a9
...@@ -185,7 +185,7 @@ def main() -> None: ...@@ -185,7 +185,7 @@ def main() -> None:
{ {
"include": f"../{task_type}", "include": f"../{task_type}",
"task": "bigbench_" + task + "_{}".format(task_type.split("_template_yaml")[0]), "task": "bigbench_" + task + "_{}".format(task_type.split("_template_yaml")[0]),
"dataset_name": task, "dataset_name": task + "_zero_shot", # zero-shot version of the dataset
}, },
f, f,
width=float("inf"), allow_unicode=True width=float("inf"), allow_unicode=True
......
group: bigbench group: bigbench
dataset_path: bigbench dataset_path: hails/bigbench
output_type: greedy_until output_type: greedy_until
training_split: train test_split: default
validation_split: validation
doc_to_text: inputs doc_to_text: inputs
doc_to_target: "{{targets[0]}}" doc_to_target: "{{targets[0]}}"
generation_kwargs: generation_kwargs:
......
group: bigbench group: bigbench
dataset_path: bigbench dataset_path: hails/bigbench
dataset_kwargs:
num_shots: 0 # TODO: num of shots should be controlled through this, not through the typical methods
# subtask_name: null
output_type: multiple_choice output_type: multiple_choice
training_split: train test_split: default
validation_split: validation
doc_to_text: inputs doc_to_text: inputs
doc_to_target: "{{multiple_choice_targets.index(targets[0])}}" doc_to_target: "{{multiple_choice_targets.index(targets[0])}}"
doc_to_choice: "{{multiple_choice_targets}}" doc_to_choice: "{{multiple_choice_targets}}"
metric_list: metric_list:
- metric: acc - metric: acc
# TODO: brier score and other metrics
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment