Importing Data

Loading recall datasets

Before computing analyses, you need to load your data into the RecallDataset format.

From HDF5

The standard format for jaxcmr data:

Code

import h5py
import jax.numpy as jnp

def load_dataset(path: str) -> dict:
    """Load a RecallDataset from HDF5."""
    with h5py.File(path, "r") as f:
        return {key: jnp.array(f[key][:]) for key in f.keys()}

dataset = load_dataset("experiment.h5")

Validating Data

Check your dataset has the required fields:

Code

required = ["subject", "listLength", "pres_itemnos", "recalls"]
for field in required:
    assert field in dataset, f"Missing required field: {field}"

See Data Format for complete field specifications.