Skip to content

ARC Challenge

get_input_data_model()

Returns ARC-Challenge input data model.

Source code in synalinks/src/datasets/built_in/arc_challenge.py 
34
35
36
37
@synalinks_export("synalinks.datasets.arc_challenge.get_input_data_model")
def get_input_data_model():
    """Returns ARC-Challenge input data model."""
    return ARCQuestion

get_output_data_model()

Returns ARC-Challenge output data model.

Source code in synalinks/src/datasets/built_in/arc_challenge.py 
40
41
42
43
@synalinks_export("synalinks.datasets.arc_challenge.get_output_data_model")
def get_output_data_model():
    """Returns ARC-Challenge output data model."""
    return ARCAnswer

iterable_dataset(repeat=1, batch_size=1, limit=None, split='train')

Streaming dataset for RL-style training.

Returns:

Type Description
HuggingFaceDataset

A streaming, iterable dataset.
Source code in synalinks/src/datasets/built_in/arc_challenge.py 
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
@synalinks_export("synalinks.datasets.arc_challenge.iterable_dataset")
def iterable_dataset(repeat=1, batch_size=1, limit=None, split="train"):
    """
    Streaming dataset for RL-style training.

    Returns:
        (HuggingFaceDataset): A streaming, iterable dataset.
    """
    return HuggingFaceDataset(
        path="allenai/ai2_arc",
        name="ARC-Challenge",
        split=split,
        streaming=True,
        input_data_model=ARCQuestion,
        input_template=_INPUT_TEMPLATE,
        output_data_model=ARCAnswer,
        output_template=_OUTPUT_TEMPLATE,
        batch_size=batch_size,
        limit=limit,
        repeat=repeat,
    )

load_data()

Load ARC-Challenge.

Returns:

Type Description
tuple

(x_train, y_train), (x_test, y_test).
Source code in synalinks/src/datasets/built_in/arc_challenge.py 
58
59
60
61
62
63
64
65
66
67
68
@synalinks_export("synalinks.datasets.arc_challenge.load_data")
def load_data():
    """
    Load ARC-Challenge.

    Returns:
        (tuple): ``(x_train, y_train), (x_test, y_test)``.
    """
    x_train, y_train = _load("train")
    x_test, y_test = _load("test")
    return (x_train, y_train), (x_test, y_test)