aiaccel.torch.datasets.RawHDF5Dataset#

class aiaccel.torch.datasets.RawHDF5Dataset(dataset_path: Path | str, grp_list: Path | str | list[str] | None = None)[source]#

A dataset class for reading data from HDF5 files.

Parameters:

  • dataset_path (Union[Path, str]) – The path to the HDF5 dataset file.

  • grp_list (Union[Path, str, List[str], None], optional) – The list of groups to load from the dataset. If None, all groups in the dataset will be loaded. If a string or Path, it should be the path to a file containing the list of groups. If a list, it should directly specify the groups to load. Defaults to None.

Raises:

NotImplementedError – If grp_list is of an unsupported type.

dataset_path#

The path to the HDF5 dataset file.

Type:

Union[Path, str]

grp_list#

The list of groups to load from the dataset.

Type:

List[str]

f#

The HDF5 file object used for reading the dataset.

Type:

Optional[h5.File]

__init__(dataset_path: Path | str, grp_list: Path | str | list[str] | None = None) → None[source]#

Methods

__init__(dataset_path[, grp_list])