Skip to content
heliaEDGE
Reference
HELIA

random_choice

Batchwise random layer choice with captured branches and owned RNG.

Machine-readable model

  • RandomChoiceclassChoose one layer for a whole batch; branches must have compatible outputs.
class

Choose one layer for a whole batch; branches must have compatible outputs.

helia_edge/layers/preprocessing/random_choice.py:9

RandomChoice(layers: list[keras.Layer], batchwise: bool = True, **kwargs={})

Choose one layer for a whole batch; branches must have compatible outputs.

Per-example layer choice was never implemented and remains unsupported. The next-major default is the supported batchwise=True mode.

Base class: BaseAugmentation.

Inherited from BaseAugmentation: augment_masks(), augment_sample(), augment_samples(), augment_targets(), get_random_transformations().