Skip to content

euroeval.dataset_configs.portuguese

[docs] module euroeval.dataset_configs.portuguese

  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
 66
 67
 68
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
"""All Portuguese dataset configurations used in EuroEval."""

from ..data_models import DatasetConfig
from ..languages import EUROPEAN_PORTUGUESE, PORTUGUESE
from ..tasks import (
    COMMON_SENSE,
    EUROPEAN_VALUES,
    INSTRUCTION_FOLLOWING,
    KNOW,
    LA,
    MCRC,
    NER,
    RC,
    SENT,
    SUMM,
)

# Official datasets ###

SST2_PT_CONFIG = DatasetConfig(
    name="sst2-pt",
    pretty_name="SST2-pt",
    source="EuroEval/sst2-pt-mini",
    task=SENT,
    languages=[PORTUGUESE, EUROPEAN_PORTUGUESE],
    labels=["positive", "negative"],
)

SCALA_PT = DatasetConfig(
    name="scala-pt",
    pretty_name="ScaLA-pt",
    source="EuroEval/scala-pt",
    task=LA,
    languages=[PORTUGUESE, EUROPEAN_PORTUGUESE],
)

HAREM_CONFIG = DatasetConfig(
    name="harem",
    pretty_name="HAREM",
    source="EuroEval/harem",
    task=NER,
    languages=[PORTUGUESE, EUROPEAN_PORTUGUESE],
)

MULTI_WIKI_QA_PT_CONFIG = DatasetConfig(
    name="multi-wiki-qa-pt",
    pretty_name="MultiWikiQA-pt",
    source="EuroEval/multi-wiki-qa-pt-pt-mini",
    task=RC,
    languages=[PORTUGUESE, EUROPEAN_PORTUGUESE],
)

PUBLICO_CONFIG = DatasetConfig(
    name="publico",
    pretty_name="Publico",
    source="EuroEval/publico-mini",
    task=SUMM,
    languages=[PORTUGUESE, EUROPEAN_PORTUGUESE],
)

MMLU_PT_CONFIG = DatasetConfig(
    name="mmlu-pt",
    pretty_name="MMLU-pt",
    source="EuroEval/mmlu-pt-mini",
    task=KNOW,
    languages=[PORTUGUESE, EUROPEAN_PORTUGUESE],
)

GOLDENSWAG_PT_CONFIG = DatasetConfig(
    name="goldenswag-pt",
    pretty_name="GoldenSwag-pt",
    source="EuroEval/goldenswag-pt-mini",
    task=COMMON_SENSE,
    languages=[PORTUGUESE, EUROPEAN_PORTUGUESE],
)

IFEVAL_PT_CONFIG = DatasetConfig(
    name="ifeval-pt",
    pretty_name="IFEval-pt",
    source="EuroEval/ifeval-pt",
    task=INSTRUCTION_FOLLOWING,
    languages=[PORTUGUESE],
    train_split=None,
    val_split=None,
)

VALEU_PT_CONFIG = DatasetConfig(
    name="valeu-pt",
    pretty_name="VaLEU-pt",
    source="EuroEval/european-values-pt",
    task=EUROPEAN_VALUES,
    languages=[PORTUGUESE, EUROPEAN_PORTUGUESE],
    train_split=None,
    val_split=None,
    bootstrap_samples=False,
    instruction_prompt="{text}",
)


# Unofficial datasets ###

BOOLQ_PT_CONFIG = DatasetConfig(
    name="boolq-pt",
    pretty_name="BoolQ-pt",
    source="EuroEval/boolq-pt",
    task=MCRC,
    languages=[PORTUGUESE, EUROPEAN_PORTUGUESE],
    unofficial=True,
)

WINOGRANDE_PT_CONFIG = DatasetConfig(
    name="winogrande-pt",
    pretty_name="Winogrande-pt",
    source="EuroEval/winogrande-pt",
    task=COMMON_SENSE,
    languages=[PORTUGUESE, EUROPEAN_PORTUGUESE],
    labels=["a", "b"],
    unofficial=True,
)