jaclearn.data.sampler#

Classes

`EpochBatchSampler`
`RenamedDictSamplerBase`
`SimpleBatchSampler`

Class EpochBatchSampler

class EpochBatchSampler[source]#

Bases: RenamedDictSamplerBase

__call__(data, keys, renames=None)[source]#: Call self as a function.

__init__(batch_size, epoch_size, rng=None)[source]#

__new__(**kwargs)#

Class RenamedDictSamplerBase

class RenamedDictSamplerBase[source]#

Bases: object

__init__()#

__new__(**kwargs)#

Class SimpleBatchSampler

class SimpleBatchSampler[source]#

Bases: RenamedDictSamplerBase

__call__(data, keys, renames=None)[source]#: Call self as a function.

__init__(batch_size, nr_repeat, rng=None)[source]#

__new__(**kwargs)#