xbatcher.loaders.keras module

class xbatcher.loaders.keras.CustomTFDataset(*args: Any, **kwargs: Any)

Bases: tensorflow.keras.utils.Sequence

__init__(X_generator, y_generator, *, transform: Optional[Callable] = None, target_transform: Optional[Callable] = None) → None

Keras Dataset adapter for Xbatcher

Parameters

X_generatorxbatcher.BatchGenerator

y_generatorxbatcher.BatchGenerator

transformcallable, optional

A function/transform that takes in an array and returns a transformed version.

target_transformcallable, optional

A function/transform that takes in the target and transforms it.