Skip to content

euroeval.dataset_configs.romanian

[docs] module euroeval.dataset_configs.romanian 

 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
65
"""All Romanian dataset configurations used in EuroEval."""

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

### Official datasets ###

ROSENT_CONFIG = DatasetConfig(
    name="ro-sent",
    pretty_name="RoSent",
    source="EuroEval/ro-sent-mini",
    task=SENT,
    languages=[ROMANIAN],
    _labels=["positive", "negative"],
)

SCALA_RO_CONFIG = DatasetConfig(
    name="scala-ro",
    pretty_name="ScaLA-ro",
    source="EuroEval/scala-ro",
    task=LA,
    languages=[ROMANIAN],
)

RONEC_CONFIG = DatasetConfig(
    name="ronec",
    pretty_name="RoNEC",
    source="EuroEval/ronec-mini",
    task=NER,
    languages=[ROMANIAN],
)

MULTI_WIKI_QA_RO_CONFIG = DatasetConfig(
    name="multi-wiki-qa-ro",
    pretty_name="MultiWikiQA-ro",
    source="EuroEval/multi-wiki-qa-ro-mini",
    task=RC,
    languages=[ROMANIAN],
)

SUMO_RO_CONFIG = DatasetConfig(
    name="sumo-ro",
    pretty_name="SumO-Ro",
    source="EuroEval/sumo-ro-mini",
    task=SUMM,
    languages=[ROMANIAN],
)

GLOBAL_MMLU_RO_CONFIG = DatasetConfig(
    name="global-mmlu-ro",
    pretty_name="GlobalMMLU-ro",
    source="EuroEval/global-mmlu-ro-mini",
    task=KNOW,
    languages=[ROMANIAN],
)

WINOGRANDE_RO_CONFIG = DatasetConfig(
    name="winogrande-ro",
    pretty_name="Winogrande-ro",
    source="EuroEval/winogrande-ro",
    task=COMMON_SENSE,
    languages=[ROMANIAN],
    _labels=["a", "b"],
)