neurai.datasets package

Subpackages

• neurai.datasets.utils_pkgs package
  • Submodules
  • Module contents

Submodules

class neurai.datasets.base.Dataset(*args, **kwds)

Bases: Generic[T_co]

An abstract class representing a Dataset. All datasets that represent a map from keys to data samples should subclass it. All subclasses should overwrite __getitem__(), supporting fetching a data sample for a given key. Subclasses could also optionally overwrite __len__(), which is expected to return the size of the dataset

Module contents