class sakkara.model.composable.base.Composable(name: str | None, group: str | Tuple[str, ...] | None, subcomponents: Dict[S, T])#

Base class for a component that can be built with underlying subcomponents

Parameters:

  • name – Name of the corresponding variable to register in PyMC.

  • group – Group of which the component is defined for.

  • members – Subset of members of column the component is defined for.

  • subcomponents – Dict of underlying ModelComponent objects.

clear()#

Clear variable and group for this and underlying components

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