IterableDataset

class braintools.trainer.IterableDataset(iterable, length=None)

Dataset wrapping an iterable.

Parameters:

• iterable (Any) – The iterable to wrap.

• length (int | None) – The length of the iterable (if known).

Examples

>>> def data_generator():
...     for i in range(100):
...         yield {'x': np.random.randn(10), 'y': i}
>>> dataset = IterableDataset(data_generator(), length=100)