xbatcher.loaders.torch module#

class xbatcher.loaders.torch.IterableDataset(*args: Any, **kwargs: Any)[source]#

Bases: torch.utils.data.IterableDataset

__init__(X_generator, y_generator) → None[source]#

PyTorch Dataset adapter for Xbatcher

Parameters
X_generatorxbatcher.BatchGenerator
y_generatorxbatcher.BatchGenerator
class xbatcher.loaders.torch.MapDataset(*args: Any, **kwargs: Any)[source]#

Bases: torch.utils.data.Dataset

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

PyTorch 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.