attribench.data.HDF5Dataset

class attribench.data.HDF5Dataset(path)[source]

Bases: Dataset

Dataset stored in a HDF5 file.

The HDF5 file must contain the following datasets:

samples: [num_samples, *sample_shape]
labels: [num_samples]

Parameters:

pathstr: Path to the HDF5 file.

Methods

Attributes

sample_shape