sfaira.data.store.batch_schedule.BatchDesignBlocks

class sfaira.data.store.batch_schedule.BatchDesignBlocks(grouping, random_access: bool, **kwargs)

Meta data-defined blocks of observations in each batch.

Note that the batches do not necessarily contain equal numbers of observations! The yielded batch size depends solely on the number of observations in a meta data partition.

Attributes

batchsplits

Batch size of the yieleded batches.

design

Yields index objects for one epoch of all data.

grouping

groups

idx

Protects property from uncontrolled changing.

n_batches

Methods