Unverified Commit 6fbebb4b authored by Angelika Romanou's avatar Angelika Romanou Committed by GitHub
Browse files

Add INCLUDE tasks (#2769)



* Add INCLUDE tasks

* pacify pre-commit

---------
Co-authored-by: default avatarBaber <baber@hey.com>
parent bb4fa95e
from functools import partial
CATEGORIES = [
"Applied Science",
"Arts & Humanities",
"Business & Commerce",
"Driving License",
"General knowledge",
"Health oriented education",
"Marine License",
"Medical License",
"Professional certification",
"STEM",
"Social Science",
]
def process_docs(dataset, category):
return dataset.filter(lambda x: x["domain"] == category)
process_functions = {
f"process_{category.lower().replace(' & ', '_').replace(' ', '_')}": partial(
process_docs, category=category
)
for category in CATEGORIES
}
globals().update(process_functions)
dataset_path: CohereForAI/include-base-44
dataset_name: Albanian
test_split: test
output_type: multiple_choice
doc_to_text: "{{question.strip()}}\nA. {{option_a}}\nB. {{option_b}}\nC. {{option_c}}\n
D. {{option_d}}\nAnswer:"
doc_to_choice:
- A
- B
- C
- D
doc_to_target: answer
metric_list:
- metric: acc
aggregation: mean
higher_is_better: true
metadata:
version: 0.0
group: include_base_44_albanian
task:
- include_base_44_albanian_few_shot_og_arts_humanities
- include_base_44_albanian_few_shot_og_stem
- include_base_44_albanian_few_shot_og_business_commerce
- include_base_44_albanian_few_shot_og_social_science
- include_base_44_albanian_few_shot_og_health_oriented_education
aggregate_metric_list:
- metric: acc
weight_by_size: true
metadata:
version: 0.0
include: _albanian_few_shot_og_template_yaml
description: Më poshtë janë pyetjet me zgjedhje të shumëfishta (me përgjigje) rreth
Arts & Humanities.
process_docs: !function 'utils.process_arts_humanities'
task: include_base_44_albanian_few_shot_og_arts_humanities
include: _albanian_few_shot_og_template_yaml
description: Më poshtë janë pyetjet me zgjedhje të shumëfishta (me përgjigje) rreth
Business & Commerce.
process_docs: !function 'utils.process_business_commerce'
task: include_base_44_albanian_few_shot_og_business_commerce
include: _albanian_few_shot_og_template_yaml
description: Më poshtë janë pyetjet me zgjedhje të shumëfishta (me përgjigje) rreth
Health oriented education.
process_docs: !function 'utils.process_health_oriented_education'
task: include_base_44_albanian_few_shot_og_health_oriented_education
include: _albanian_few_shot_og_template_yaml
description: Më poshtë janë pyetjet me zgjedhje të shumëfishta (me përgjigje) rreth
Social Science.
process_docs: !function 'utils.process_social_science'
task: include_base_44_albanian_few_shot_og_social_science
include: _albanian_few_shot_og_template_yaml
description: Më poshtë janë pyetjet me zgjedhje të shumëfishta (me përgjigje) rreth
STEM.
process_docs: !function 'utils.process_stem'
task: include_base_44_albanian_few_shot_og_stem
from functools import partial
CATEGORIES = [
"Applied Science",
"Arts & Humanities",
"Business & Commerce",
"Driving License",
"General knowledge",
"Health oriented education",
"Marine License",
"Medical License",
"Professional certification",
"STEM",
"Social Science",
]
def process_docs(dataset, category):
return dataset.filter(lambda x: x["domain"] == category)
process_functions = {
f"process_{category.lower().replace(' & ', '_').replace(' ', '_')}": partial(
process_docs, category=category
)
for category in CATEGORIES
}
globals().update(process_functions)
dataset_path: CohereForAI/include-base-44
dataset_name: Arabic
test_split: test
output_type: multiple_choice
doc_to_text: "{{question.strip()}}\nA. {{option_a}}\nB. {{option_b}}\nC. {{option_c}}\n
D. {{option_d}}\nAnswer:"
doc_to_choice:
- A
- B
- C
- D
doc_to_target: answer
metric_list:
- metric: acc
aggregation: mean
higher_is_better: true
metadata:
version: 0.0
group: include_base_44_arabic
task:
- include_base_44_arabic_few_shot_og_arts_humanities
- include_base_44_arabic_few_shot_og_stem
- include_base_44_arabic_few_shot_og_social_science
- include_base_44_arabic_few_shot_og_driving_license
- include_base_44_arabic_few_shot_og_general_knowledge
- include_base_44_arabic_few_shot_og_business_commerce
aggregate_metric_list:
- metric: acc
weight_by_size: true
metadata:
version: 0.0
include: _arabic_few_shot_og_template_yaml
description: فيما يلي أسئلة اختيارية متعددة (مع الإجابات) حول Arts & Humanities.
process_docs: !function 'utils.process_arts_humanities'
task: include_base_44_arabic_few_shot_og_arts_humanities
include: _arabic_few_shot_og_template_yaml
description: فيما يلي أسئلة اختيارية متعددة (مع الإجابات) حول Business & Commerce.
process_docs: !function 'utils.process_business_commerce'
task: include_base_44_arabic_few_shot_og_business_commerce
include: _arabic_few_shot_og_template_yaml
description: فيما يلي أسئلة اختيارية متعددة (مع الإجابات) حول Driving License.
process_docs: !function 'utils.process_driving_license'
task: include_base_44_arabic_few_shot_og_driving_license
include: _arabic_few_shot_og_template_yaml
description: فيما يلي أسئلة اختيارية متعددة (مع الإجابات) حول General knowledge.
process_docs: !function 'utils.process_general_knowledge'
task: include_base_44_arabic_few_shot_og_general_knowledge
include: _arabic_few_shot_og_template_yaml
description: فيما يلي أسئلة اختيارية متعددة (مع الإجابات) حول Social Science.
process_docs: !function 'utils.process_social_science'
task: include_base_44_arabic_few_shot_og_social_science
include: _arabic_few_shot_og_template_yaml
description: فيما يلي أسئلة اختيارية متعددة (مع الإجابات) حول STEM.
process_docs: !function 'utils.process_stem'
task: include_base_44_arabic_few_shot_og_stem
from functools import partial
CATEGORIES = [
"Applied Science",
"Arts & Humanities",
"Business & Commerce",
"Driving License",
"General knowledge",
"Health oriented education",
"Marine License",
"Medical License",
"Professional certification",
"STEM",
"Social Science",
]
def process_docs(dataset, category):
return dataset.filter(lambda x: x["domain"] == category)
process_functions = {
f"process_{category.lower().replace(' & ', '_').replace(' ', '_')}": partial(
process_docs, category=category
)
for category in CATEGORIES
}
globals().update(process_functions)
dataset_path: CohereForAI/include-base-44
dataset_name: Armenian
test_split: test
output_type: multiple_choice
doc_to_text: "{{question.strip()}}\nA. {{option_a}}\nB. {{option_b}}\nC. {{option_c}}\n
D. {{option_d}}\nAnswer:"
doc_to_choice:
- A
- B
- C
- D
doc_to_target: answer
metric_list:
- metric: acc
aggregation: mean
higher_is_better: true
metadata:
version: 0.0
group: include_base_44_armenian
task:
- include_base_44_armenian_few_shot_og_driving_license
- include_base_44_armenian_few_shot_og_social_science
- include_base_44_armenian_few_shot_og_arts_humanities
- include_base_44_armenian_few_shot_og_stem
aggregate_metric_list:
- metric: acc
weight_by_size: true
metadata:
version: 0.0
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