Dataset

jax_dataloader.datasets.Dataset

A pytorch-like Dataset class.

Source code in jax_dataloader/datasets.py

class Dataset:
    """A pytorch-like Dataset class."""

    def __len__(self):
        raise NotImplementedError

    def __getitem__(self, index):
        raise NotImplementedError

Functions

__getitem__(index)

Source code in jax_dataloader/datasets.py

def __getitem__(self, index):
    raise NotImplementedError

__len__()

Source code in jax_dataloader/datasets.py

def __len__(self):
    raise NotImplementedError