uqlm.utils.dataloader.load_example_dataset#

uqlm.utils.dataloader.load_example_dataset(name, n=None, cols=None)#

Load a dataset for testing purposes.

Return type:

DataFrame

Parameters:

name (str) – The name of the dataset to load. Must be one of “svamp”, “gsm8k”, “ai2_arc”, “csqa”, “nq_open”, “popqa”
n (int, optional) – Number of rows to load from the dataset.

Returns:

The loaded dataset.

Return type:

pd.DataFrame

Example

>>> from uqlm.utils.dataloader import load_example_dataset
>>> df = load_example_dataset("gsm8k", n=1000)
>>> df.shape
(1000, 2)