Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
gaoqiong
lm-evaluation-harness
Commits
e58b8182
Commit
e58b8182
authored
Aug 08, 2024
by
lintangsutawika
Browse files
resolved merge conflict
parents
d213a533
0571eeb1
Changes
105
Hide whitespace changes
Inline
Side-by-side
Showing
20 changed files
with
93 additions
and
11 deletions
+93
-11
lm_eval/tasks/afrimmlu/translate/afrimmlu_translate_fra.yaml
lm_eval/tasks/afrimmlu/translate/afrimmlu_translate_fra.yaml
+4
-0
lm_eval/tasks/afrimmlu/translate/afrimmlu_translate_hau.yaml
lm_eval/tasks/afrimmlu/translate/afrimmlu_translate_hau.yaml
+4
-0
lm_eval/tasks/afrimmlu/translate/afrimmlu_translate_ibo.yaml
lm_eval/tasks/afrimmlu/translate/afrimmlu_translate_ibo.yaml
+4
-0
lm_eval/tasks/afrimmlu/translate/afrimmlu_translate_kin.yaml
lm_eval/tasks/afrimmlu/translate/afrimmlu_translate_kin.yaml
+4
-0
lm_eval/tasks/afrimmlu/translate/afrimmlu_translate_lin.yaml
lm_eval/tasks/afrimmlu/translate/afrimmlu_translate_lin.yaml
+4
-0
lm_eval/tasks/afrimmlu/translate/afrimmlu_translate_lug.yaml
lm_eval/tasks/afrimmlu/translate/afrimmlu_translate_lug.yaml
+4
-0
lm_eval/tasks/afrimmlu/translate/afrimmlu_translate_orm.yaml
lm_eval/tasks/afrimmlu/translate/afrimmlu_translate_orm.yaml
+4
-0
lm_eval/tasks/afrimmlu/translate/afrimmlu_translate_sna.yaml
lm_eval/tasks/afrimmlu/translate/afrimmlu_translate_sna.yaml
+4
-0
lm_eval/tasks/afrimmlu/translate/afrimmlu_translate_sot.yaml
lm_eval/tasks/afrimmlu/translate/afrimmlu_translate_sot.yaml
+4
-0
lm_eval/tasks/afrimmlu/translate/afrimmlu_translate_swa.yaml
lm_eval/tasks/afrimmlu/translate/afrimmlu_translate_swa.yaml
+4
-0
lm_eval/tasks/afrimmlu/translate/afrimmlu_translate_twi.yaml
lm_eval/tasks/afrimmlu/translate/afrimmlu_translate_twi.yaml
+4
-0
lm_eval/tasks/afrimmlu/translate/afrimmlu_translate_wol.yaml
lm_eval/tasks/afrimmlu/translate/afrimmlu_translate_wol.yaml
+4
-0
lm_eval/tasks/afrimmlu/translate/afrimmlu_translate_xho.yaml
lm_eval/tasks/afrimmlu/translate/afrimmlu_translate_xho.yaml
+4
-0
lm_eval/tasks/afrimmlu/translate/afrimmlu_translate_yor.yaml
lm_eval/tasks/afrimmlu/translate/afrimmlu_translate_yor.yaml
+4
-0
lm_eval/tasks/afrimmlu/translate/afrimmlu_translate_zul.yaml
lm_eval/tasks/afrimmlu/translate/afrimmlu_translate_zul.yaml
+4
-0
lm_eval/tasks/afrimmlu/translate/utils.py
lm_eval/tasks/afrimmlu/translate/utils.py
+4
-1
lm_eval/tasks/afrimmlu/utils.py
lm_eval/tasks/afrimmlu/utils.py
+4
-1
lm_eval/tasks/afrixnli/_afrixnli.yaml
lm_eval/tasks/afrixnli/_afrixnli.yaml
+12
-0
lm_eval/tasks/afrixnli/_afrixnli_ablations.yaml
lm_eval/tasks/afrixnli/_afrixnli_ablations.yaml
+12
-0
lm_eval/tasks/afrixnli/anli_prompt/en-direct/utils.py
lm_eval/tasks/afrixnli/anli_prompt/en-direct/utils.py
+1
-9
No files found.
lm_eval/tasks/afrimmlu/translate/afrimmlu_translate_fra.yaml
View file @
e58b8182
dataset_name
:
fra
<<<<<<< HEAD
include
:
_translate_yaml
=======
include
:
afrimmlu_common_translate_yaml
>
>>>>>>
0571eeb14d4e48aac51956a726c62cd8b382b3d8
t
ask: afrimmlu_translate_fra
lm_eval/tasks/afrimmlu/translate/afrimmlu_translate_hau.yaml
View file @
e58b8182
dataset_name
:
hau
<<<<<<< HEAD
include
:
_translate_yaml
=======
include
:
afrimmlu_common_translate_yaml
>
>>>>>>
0571eeb14d4e48aac51956a726c62cd8b382b3d8
t
ask: afrimmlu_translate_hau
lm_eval/tasks/afrimmlu/translate/afrimmlu_translate_ibo.yaml
View file @
e58b8182
dataset_name
:
ibo
<<<<<<< HEAD
include
:
_translate_yaml
=======
include
:
afrimmlu_common_translate_yaml
>
>>>>>>
0571eeb14d4e48aac51956a726c62cd8b382b3d8
t
ask: afrimmlu_translate_ibo
lm_eval/tasks/afrimmlu/translate/afrimmlu_translate_kin.yaml
View file @
e58b8182
dataset_name
:
kin
<<<<<<< HEAD
include
:
_translate_yaml
=======
include
:
afrimmlu_common_translate_yaml
>
>>>>>>
0571eeb14d4e48aac51956a726c62cd8b382b3d8
t
ask: afrimmlu_translate_kin
lm_eval/tasks/afrimmlu/translate/afrimmlu_translate_lin.yaml
View file @
e58b8182
dataset_name
:
lin
<<<<<<< HEAD
include
:
_translate_yaml
=======
include
:
afrimmlu_common_translate_yaml
>
>>>>>>
0571eeb14d4e48aac51956a726c62cd8b382b3d8
t
ask: afrimmlu_translate_lin
lm_eval/tasks/afrimmlu/translate/afrimmlu_translate_lug.yaml
View file @
e58b8182
dataset_name
:
lug
<<<<<<< HEAD
include
:
_translate_yaml
=======
include
:
afrimmlu_common_translate_yaml
>
>>>>>>
0571eeb14d4e48aac51956a726c62cd8b382b3d8
t
ask: afrimmlu_translate_lug
lm_eval/tasks/afrimmlu/translate/afrimmlu_translate_orm.yaml
View file @
e58b8182
dataset_name
:
orm
<<<<<<< HEAD
include
:
_translate_yaml
=======
include
:
afrimmlu_common_translate_yaml
>
>>>>>>
0571eeb14d4e48aac51956a726c62cd8b382b3d8
t
ask: afrimmlu_translate_orm
lm_eval/tasks/afrimmlu/translate/afrimmlu_translate_sna.yaml
View file @
e58b8182
dataset_name
:
sna
<<<<<<< HEAD
include
:
_translate_yaml
=======
include
:
afrimmlu_common_translate_yaml
>
>>>>>>
0571eeb14d4e48aac51956a726c62cd8b382b3d8
t
ask: afrimmlu_translate_sna
lm_eval/tasks/afrimmlu/translate/afrimmlu_translate_sot.yaml
View file @
e58b8182
dataset_name
:
sot
<<<<<<< HEAD
include
:
_translate_yaml
=======
include
:
afrimmlu_common_translate_yaml
>
>>>>>>
0571eeb14d4e48aac51956a726c62cd8b382b3d8
t
ask: afrimmlu_translate_sot
lm_eval/tasks/afrimmlu/translate/afrimmlu_translate_swa.yaml
View file @
e58b8182
dataset_name
:
swa
<<<<<<< HEAD
include
:
_translate_yaml
=======
include
:
afrimmlu_common_translate_yaml
>
>>>>>>
0571eeb14d4e48aac51956a726c62cd8b382b3d8
t
ask: afrimmlu_translate_swa
lm_eval/tasks/afrimmlu/translate/afrimmlu_translate_twi.yaml
View file @
e58b8182
dataset_name
:
twi
<<<<<<< HEAD
include
:
_translate_yaml
=======
include
:
afrimmlu_common_translate_yaml
>
>>>>>>
0571eeb14d4e48aac51956a726c62cd8b382b3d8
t
ask: afrimmlu_translate_twi
lm_eval/tasks/afrimmlu/translate/afrimmlu_translate_wol.yaml
View file @
e58b8182
dataset_name
:
wol
<<<<<<< HEAD
include
:
_translate_yaml
=======
include
:
afrimmlu_common_translate_yaml
>
>>>>>>
0571eeb14d4e48aac51956a726c62cd8b382b3d8
t
ask: afrimmlu_translate_wol
lm_eval/tasks/afrimmlu/translate/afrimmlu_translate_xho.yaml
View file @
e58b8182
dataset_name
:
xho
<<<<<<< HEAD
include
:
_translate_yaml
=======
include
:
afrimmlu_common_translate_yaml
>
>>>>>>
0571eeb14d4e48aac51956a726c62cd8b382b3d8
t
ask: afrimmlu_translate_xho
lm_eval/tasks/afrimmlu/translate/afrimmlu_translate_yor.yaml
View file @
e58b8182
dataset_name
:
yor
<<<<<<< HEAD
include
:
_translate_yaml
=======
include
:
afrimmlu_common_translate_yaml
>
>>>>>>
0571eeb14d4e48aac51956a726c62cd8b382b3d8
t
ask: afrimmlu_translate_yor
lm_eval/tasks/afrimmlu/translate/afrimmlu_translate_zul.yaml
View file @
e58b8182
dataset_name
:
zul
<<<<<<< HEAD
include
:
_translate_yaml
=======
include
:
afrimmlu_common_translate_yaml
>
>>>>>>
0571eeb14d4e48aac51956a726c62cd8b382b3d8
t
ask: afrimmlu_translate_zul
lm_eval/tasks/afrimmlu/translate/utils.py
View file @
e58b8182
from
sklearn.metrics
import
f1_score
from
lm_eval.utils
import
weighted_
f1_score
def
doc_to_choice
(
doc
):
...
...
@@ -30,6 +30,7 @@ def doc_to_text(doc):
choice4
=
choices
[
3
],
)
return
text
<<<<<<<
HEAD
def
weighted_f1_score
(
items
):
...
...
@@ -38,3 +39,5 @@ def weighted_f1_score(items):
preds
=
unzipped_list
[
1
]
fscore
=
f1_score
(
golds
,
preds
,
average
=
"weighted"
)
return
fscore
=======
>>>>>>>
0571
eeb14d4e48aac51956a726c62cd8b382b3d8
lm_eval/tasks/afrimmlu/utils.py
View file @
e58b8182
from
sklearn.metrics
import
f1_score
from
lm_eval.utils
import
weighted_
f1_score
def
doc_to_choice
(
doc
):
...
...
@@ -30,6 +30,7 @@ def doc_to_text(doc):
choice4
=
choices
[
3
],
)
return
text
<<<<<<<
HEAD
def
weighted_f1_score
(
items
):
...
...
@@ -38,3 +39,5 @@ def weighted_f1_score(items):
preds
=
unzipped_list
[
1
]
fscore
=
f1_score
(
golds
,
preds
,
average
=
"weighted"
)
return
fscore
=======
>>>>>>>
0571
eeb14d4e48aac51956a726c62cd8b382b3d8
lm_eval/tasks/afrixnli/_afrixnli.yaml
0 → 100644
View file @
e58b8182
group
:
afrixnli
group_alias
:
AfriXNLI
task
:
-
group
:
afrixnli_direct
group_alias
:
Prompt in African Languages
task
:
-
afrixnli_native_direct
-
afrixnli_native_direct_eng_only
-
group
:
afrixnli_lai_prompt_translate
group_alias
:
Translate-Test
task
:
-
afrixnli_lai_prompt_translate
\ No newline at end of file
lm_eval/tasks/afrixnli/_afrixnli_ablations.yaml
0 → 100644
View file @
e58b8182
group
:
afrixnli
group_alias
:
AfriXNLI
task
:
-
group
:
afrixnli_direct
group_alias
:
Prompt in African Languages
task
:
-
afrixnli_native_direct
-
afrixnli_native_direct_eng_only
-
group
:
afrixnli_translate
group_alias
:
Translate-Test
task
:
-
afrixnli_translate
\ No newline at end of file
lm_eval/tasks/afrixnli/anli_prompt/en-direct/utils.py
View file @
e58b8182
from
sklearn.metrics
import
f1_score
from
lm_eval.utils
import
weighted_
f1_score
def
doc_to_target
(
doc
):
replacements
=
{
0
:
"True"
,
1
:
"Neither"
,
2
:
"False"
}
return
replacements
[
doc
[
"label"
]]
def
weighted_f1_score
(
items
):
unzipped_list
=
list
(
zip
(
*
items
))
golds
=
unzipped_list
[
0
]
preds
=
unzipped_list
[
1
]
fscore
=
f1_score
(
golds
,
preds
,
average
=
"weighted"
)
return
fscore
Prev
1
2
3
4
5
6
Next
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment