Skip to content

euroeval.dataset_configs.albanian

[docs] module euroeval.dataset_configs.albanian

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
"""All Albanian dataset configurations used in EuroEval."""

from ..data_models import DatasetConfig
from ..languages import ALBANIAN
from ..tasks import COMMON_SENSE, KNOW, LA, NER, RC, SENT, SUMM

### Official datasets ###

MMS_SQ_CONFIG = DatasetConfig(
    name="mms-sq",
    pretty_name="MMS-sq",
    source="EuroEval/mms-sq-mini",
    task=SENT,
    languages=[ALBANIAN],
)

SCALA_SQ_CONFIG = DatasetConfig(
    name="scala-sq",
    pretty_name="ScaLA-sq",
    source="EuroEval/scala-sq",
    task=LA,
    languages=[ALBANIAN],
)

WIKIANN_SQ_CONFIG = DatasetConfig(
    name="wikiann-sq",
    pretty_name="WikiANN-sq",
    source="EuroEval/wikiann-sq-mini",
    task=NER,
    languages=[ALBANIAN],
)

MULTI_WIKI_QA_SQ_CONFIG = DatasetConfig(
    name="multi-wiki-qa-sq",
    pretty_name="MultiWikiQA-sq",
    source="EuroEval/multi-wiki-qa-sq-mini",
    task=RC,
    languages=[ALBANIAN],
)

LR_SUM_SQ_CONFIG = DatasetConfig(
    name="lr-sum-sq",
    pretty_name="LRSum-sq",
    source="EuroEval/lr-sum-sq-mini",
    task=SUMM,
    languages=[ALBANIAN],
)

GLOBAL_MMLU_LITE_SQ_CONFIG = DatasetConfig(
    name="global-mmlu-lite-sq",
    pretty_name="GlobalMMLULite-sq",
    source="EuroEval/global-mmlu-lite-sq",
    task=KNOW,
    languages=[ALBANIAN],
)

WINOGRANDE_SQ_CONFIG = DatasetConfig(
    name="winogrande-sq",
    pretty_name="Winogrande-sq",
    source="EuroEval/winogrande-sq",
    task=COMMON_SENSE,
    languages=[ALBANIAN],
    _labels=["a", "b"],
)