Merge branch 'main' into feature/eval_from_config

lm_eval/tasks/afrixnli/direct/prompt_3/afrixnli_orm.yaml

+# Generated by utils.py
+dataset_name: orm
+doc_to_text: "Given the following premise and hypothesis in Oromo, identify if the\
+  \ premise entails, contradicts, or is neutral towards the hypothesis. Please respond\
+  \ with exact 'entailment', 'contradiction', or 'neutral'. \n\nPremise: {{premise}}\
+  \ \nHypothesis: {{hypothesis}}"
+include: afrixnli_yaml
+task: afrixnli_orm_prompt_3

lm_eval/tasks/afrixnli/direct/prompt_3/afrixnli_sna.yaml

+# Generated by utils.py
+dataset_name: sna
+doc_to_text: "Given the following premise and hypothesis in chiShona, identify if\
+  \ the premise entails, contradicts, or is neutral towards the hypothesis. Please\
+  \ respond with exact 'entailment', 'contradiction', or 'neutral'. \n\nPremise: {{premise}}\
+  \ \nHypothesis: {{hypothesis}}"
+include: afrixnli_yaml
+task: afrixnli_sna_prompt_3

lm_eval/tasks/afrixnli/direct/prompt_3/afrixnli_sot.yaml

+# Generated by utils.py
+dataset_name: sot
+doc_to_text: "Given the following premise and hypothesis in Sesotho, identify if the\
+  \ premise entails, contradicts, or is neutral towards the hypothesis. Please respond\
+  \ with exact 'entailment', 'contradiction', or 'neutral'. \n\nPremise: {{premise}}\
+  \ \nHypothesis: {{hypothesis}}"
+include: afrixnli_yaml
+task: afrixnli_sot_prompt_3

lm_eval/tasks/afrixnli/direct/prompt_3/afrixnli_swa.yaml

+# Generated by utils.py
+dataset_name: swa
+doc_to_text: "Given the following premise and hypothesis in Swahili, identify if the\
+  \ premise entails, contradicts, or is neutral towards the hypothesis. Please respond\
+  \ with exact 'entailment', 'contradiction', or 'neutral'. \n\nPremise: {{premise}}\
+  \ \nHypothesis: {{hypothesis}}"
+include: afrixnli_yaml
+task: afrixnli_swa_prompt_3

lm_eval/tasks/afrixnli/direct/prompt_3/afrixnli_twi.yaml

+# Generated by utils.py
+dataset_name: twi
+doc_to_text: "Given the following premise and hypothesis in Twi, identify if the premise\
+  \ entails, contradicts, or is neutral towards the hypothesis. Please respond with\
+  \ exact 'entailment', 'contradiction', or 'neutral'. \n\nPremise: {{premise}} \n\
+  Hypothesis: {{hypothesis}}"
+include: afrixnli_yaml
+task: afrixnli_twi_prompt_3

lm_eval/tasks/afrixnli/direct/prompt_3/afrixnli_wol.yaml

+# Generated by utils.py
+dataset_name: wol
+doc_to_text: "Given the following premise and hypothesis in Wolof, identify if the\
+  \ premise entails, contradicts, or is neutral towards the hypothesis. Please respond\
+  \ with exact 'entailment', 'contradiction', or 'neutral'. \n\nPremise: {{premise}}\
+  \ \nHypothesis: {{hypothesis}}"
+include: afrixnli_yaml
+task: afrixnli_wol_prompt_3

lm_eval/tasks/afrixnli/direct/prompt_3/afrixnli_xho.yaml

+# Generated by utils.py
+dataset_name: xho
+doc_to_text: "Given the following premise and hypothesis in isiXhosa, identify if\
+  \ the premise entails, contradicts, or is neutral towards the hypothesis. Please\
+  \ respond with exact 'entailment', 'contradiction', or 'neutral'. \n\nPremise: {{premise}}\
+  \ \nHypothesis: {{hypothesis}}"
+include: afrixnli_yaml
+task: afrixnli_xho_prompt_3

lm_eval/tasks/afrixnli/direct/prompt_3/afrixnli_yaml

+tag:
+    - afrixnli_tasks
+    - afrixnli_tasks_prompt_3
+dataset_path: masakhane/afrixnli
+dataset_name: null
+output_type: multiple_choice
+validation_split: validation
+test_split: test
+fewshot_split: validation
+doc_to_target: !function utils.doc_to_target
+doc_to_choice:
+  - "entailment"
+  - "neutral"
+  - "contradiction"
+should_decontaminate: true
+doc_to_decontamination_query: premise
+metric_list:
+  - metric: f1
+    aggregation: !function utils.weighted_f1_score
+    average: weighted
+    higher_is_better: True
+    ignore_case: true
+    ignore_punctuation: true
+  - metric: acc
+    aggregation: mean
+    higher_is_better: true
+    ignore_case: true
+    ignore_punctuation: true
+metadata:
+  version: 1.0

lm_eval/tasks/afrixnli/direct/prompt_3/afrixnli_yor.yaml

+# Generated by utils.py
+dataset_name: yor
+doc_to_text: "Given the following premise and hypothesis in Yoruba, identify if the\
+  \ premise entails, contradicts, or is neutral towards the hypothesis. Please respond\
+  \ with exact 'entailment', 'contradiction', or 'neutral'. \n\nPremise: {{premise}}\
+  \ \nHypothesis: {{hypothesis}}"
+include: afrixnli_yaml
+task: afrixnli_yor_prompt_3

lm_eval/tasks/afrixnli/direct/prompt_3/afrixnli_zul.yaml

+# Generated by utils.py
+dataset_name: zul
+doc_to_text: "Given the following premise and hypothesis in Zulu, identify if the\
+  \ premise entails, contradicts, or is neutral towards the hypothesis. Please respond\
+  \ with exact 'entailment', 'contradiction', or 'neutral'. \n\nPremise: {{premise}}\
+  \ \nHypothesis: {{hypothesis}}"
+include: afrixnli_yaml
+task: afrixnli_zul_prompt_3

lm_eval/tasks/afrixnli/direct/prompt_3/utils.py

+from lm_eval.utils import weighted_f1_score
+
+
+def doc_to_target(doc):
+    replacements = {0: "entailment", 1: "neutral", 2: "contradiction"}
+    return replacements[doc["label"]]

lm_eval/tasks/afrixnli/direct/prompt_4/afrixnli_amh.yaml

+# Generated by utils.py
+dataset_name: amh
+doc_to_text: "You are an expert in Natural Language Inference (NLI) specializing in\
+  \ the Amharic language.\nAnalyze the premise and hypothesis given in Amharic, and\
+  \ determine the relationship between them.\n Respond with one of the following options:\
+  \ 'entailment', 'contradiction', or 'neutral'. \n\nPremise: {{premise}} \nHypothesis:\
+  \ {{hypothesis}}"
+include: afrixnli_yaml
+task: afrixnli_amh_prompt_4

lm_eval/tasks/afrixnli/direct/prompt_4/afrixnli_eng.yaml

+# Generated by utils.py
+dataset_name: eng
+doc_to_text: "You are an expert in Natural Language Inference (NLI) specializing in\
+  \ the English language.\nAnalyze the premise and hypothesis given in English, and\
+  \ determine the relationship between them.\n Respond with one of the following options:\
+  \ 'entailment', 'contradiction', or 'neutral'. \n\nPremise: {{premise}} \nHypothesis:\
+  \ {{hypothesis}}"
+include: afrixnli_yaml
+task: afrixnli_eng_prompt_4

lm_eval/tasks/afrixnli/direct/prompt_4/afrixnli_ewe.yaml

+# Generated by utils.py
+dataset_name: ewe
+doc_to_text: "You are an expert in Natural Language Inference (NLI) specializing in\
+  \ the Ewe language.\nAnalyze the premise and hypothesis given in Ewe, and determine\
+  \ the relationship between them.\n Respond with one of the following options: 'entailment',\
+  \ 'contradiction', or 'neutral'. \n\nPremise: {{premise}} \nHypothesis: {{hypothesis}}"
+include: afrixnli_yaml
+task: afrixnli_ewe_prompt_4

lm_eval/tasks/afrixnli/direct/prompt_4/afrixnli_fra.yaml

+# Generated by utils.py
+dataset_name: fra
+doc_to_text: "You are an expert in Natural Language Inference (NLI) specializing in\
+  \ the French language.\nAnalyze the premise and hypothesis given in French, and\
+  \ determine the relationship between them.\n Respond with one of the following options:\
+  \ 'entailment', 'contradiction', or 'neutral'. \n\nPremise: {{premise}} \nHypothesis:\
+  \ {{hypothesis}}"
+include: afrixnli_yaml
+task: afrixnli_fra_prompt_4

lm_eval/tasks/afrixnli/direct/prompt_4/afrixnli_hau.yaml

+# Generated by utils.py
+dataset_name: hau
+doc_to_text: "You are an expert in Natural Language Inference (NLI) specializing in\
+  \ the Hausa language.\nAnalyze the premise and hypothesis given in Hausa, and determine\
+  \ the relationship between them.\n Respond with one of the following options: 'entailment',\
+  \ 'contradiction', or 'neutral'. \n\nPremise: {{premise}} \nHypothesis: {{hypothesis}}"
+include: afrixnli_yaml
+task: afrixnli_hau_prompt_4

lm_eval/tasks/afrixnli/direct/prompt_4/afrixnli_ibo.yaml

+# Generated by utils.py
+dataset_name: ibo
+doc_to_text: "You are an expert in Natural Language Inference (NLI) specializing in\
+  \ the Igbo language.\nAnalyze the premise and hypothesis given in Igbo, and determine\
+  \ the relationship between them.\n Respond with one of the following options: 'entailment',\
+  \ 'contradiction', or 'neutral'. \n\nPremise: {{premise}} \nHypothesis: {{hypothesis}}"
+include: afrixnli_yaml
+task: afrixnli_ibo_prompt_4

lm_eval/tasks/afrixnli/direct/prompt_4/afrixnli_kin.yaml

+# Generated by utils.py
+dataset_name: kin
+doc_to_text: "You are an expert in Natural Language Inference (NLI) specializing in\
+  \ the Kinyarwanda language.\nAnalyze the premise and hypothesis given in Kinyarwanda,\
+  \ and determine the relationship between them.\n Respond with one of the following\
+  \ options: 'entailment', 'contradiction', or 'neutral'. \n\nPremise: {{premise}}\
+  \ \nHypothesis: {{hypothesis}}"
+include: afrixnli_yaml
+task: afrixnli_kin_prompt_4

lm_eval/tasks/afrixnli/direct/prompt_4/afrixnli_lin.yaml

+# Generated by utils.py
+dataset_name: lin
+doc_to_text: "You are an expert in Natural Language Inference (NLI) specializing in\
+  \ the Lingala language.\nAnalyze the premise and hypothesis given in Lingala, and\
+  \ determine the relationship between them.\n Respond with one of the following options:\
+  \ 'entailment', 'contradiction', or 'neutral'. \n\nPremise: {{premise}} \nHypothesis:\
+  \ {{hypothesis}}"
+include: afrixnli_yaml
+task: afrixnli_lin_prompt_4

lm_eval/tasks/afrixnli/direct/prompt_4/afrixnli_lug.yaml

+# Generated by utils.py
+dataset_name: lug
+doc_to_text: "You are an expert in Natural Language Inference (NLI) specializing in\
+  \ the Luganda language.\nAnalyze the premise and hypothesis given in Luganda, and\
+  \ determine the relationship between them.\n Respond with one of the following options:\
+  \ 'entailment', 'contradiction', or 'neutral'. \n\nPremise: {{premise}} \nHypothesis:\
+  \ {{hypothesis}}"
+include: afrixnli_yaml
+task: afrixnli_lug_prompt_4