class sakkara.model.minibatch.MinibatchComponent(component: ModelComponent, batch_size: int, group: str)#

Wrapper component for mini-batching of any (repeatable component)

Parameters:
  • component – Component to be wrapped.

  • batch_size – Batch size of the mini-batch.

  • group – Group to apply mini-batching on.

build_representation(groupset: GroupSet) None#

Build the group of this component, performed after prebuild.

Parameters:

groupset – Groups to be used for building all components of the model.

build_variable() None#

Build the variable, performed after prebuild and build_group.

clear() None#

Clear variable and group for this and underlying components

prebuild(groupset: GroupSet) None#

All operations to be performed before building group and variable, e.g., building the underlying components.

Parameters:

groupset – Groups to be used for building all components of the model.

retrieve_groups() Set[str]#

Retrieve group names for this and underlying components

to_minibatch(batch_size: int, group: str) ModelComponent#

Convert the component to counterpart suitable for mini-batches