trw.transforms.transforms_random_cutout

Module Contents

Classes

TransformRandomCutout

Randomly flip the axis of selected features

Functions

_transform_random_cutout(feature_names, batch, cutout_size, cutout_value_fn, probability)

trw.transforms.transforms_random_cutout._transform_random_cutout(feature_names, batch, cutout_size, cutout_value_fn, probability)

class trw.transforms.transforms_random_cutout.TransformRandomCutout(cutout_size: Union[trw.basic_typing.ShapeCX, Callable[[], trw.basic_typing.ShapeCX]], criteria_fn: Optional[trw.transforms.transforms.CriteriaFn] = None, probability: float = 1.0, cutout_value_fn: Callable[[trw.basic_typing.TensorNCX], None] = functools.partial(cutout_function.cutout_value_fn_constant, value=0))

Bases: trw.transforms.transforms.TransformBatchWithCriteria

Randomly flip the axis of selected features