deepr.prepros.Batch

class deepr.prepros.Batch(batch_size, drop_remainder=False)

Combines consecutive elements of a dataset into batches.

count

Number of dataset repeat

Type:

int

modes

Active modes for the map (will skip modes not in modes). Default is None (all modes are considered active modes).

Type:

Iterable[str], Optional

__init__(batch_size, drop_remainder=False)

Methods

__init__(batch_size[, drop_remainder])
apply(dataset[, mode])	Pre-process a dataset