trw.transforms.transforms_random_flip

Module Contents

Classes

TransformRandomFlip

Randomly flip the axis of selected features

Functions

_transform_random_flip(feature_names, batch, axis, flip_probability)

trw.transforms.transforms_random_flip._transform_random_flip(feature_names, batch, axis, flip_probability)

class trw.transforms.transforms_random_flip.TransformRandomFlip(axis: int, flip_probability: float = 0.5, criteria_fn: Optional[trw.transforms.transforms.CriteriaFn] = None)

Bases: trw.transforms.transforms.TransformBatchWithCriteria

Randomly flip the axis of selected features