xbatcher.loaders.keras.CustomTFDataset#
- class xbatcher.loaders.keras.CustomTFDataset(*args: Any, **kwargs: Any)[source]#
- __init__(X_generator, y_generator, *, transform: Optional[Callable] = None, target_transform: Optional[Callable] = None) None [source]#
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.
Methods
__init__
(X_generator, y_generator, *[, ...])Keras Dataset adapter for Xbatcher