DataLoaderBase

class DataLoaderBase(batch_size: int = 32, shuffle: bool = True)

Bases: bluebird.data.DataIterator

Base class that every data loader inherits.

Example:

class CustomDataLoader(DataLoaderBase):
    def __len__(self) -> int:
        return len(x_train)

    def __getitem__(self, idx) -> int:
        x = x_train[idx: idx+self.batch_size]
        y = y_train[idx: idx+self.batch_size]

        return Batch(x, y)

__init__(batch_size: int = 32, shuffle: bool = True)

Initalizes the object.

Parameters

• batch_size (int) – length of every batch size, defaults to 32

• shuffle (bool) – shuffles data if true, defaults to True

Methods

__init__([batch_size, shuffle])

Initalizes the object.