trw.transforms.transforms

Module Contents

Classes

Transform	Abstraction of a batch transform
TransformBatchWithCriteria	Helper function to apply a given transform function on features that satisfy a criteria
TransformBatchJointWithCriteria	Helper function to apply a given transform function on features that satisfy a criteria. Feature will be jointly transformed

Functions

criteria_is_array_3_or_above(feature_name, feature_value)	Return True if the feature is a numpy or torch array dim >= 3
criteria_feature_name(feature_name, feature_value, feature_names)	Return True if the feature name belongs to a given set of names

class trw.transforms.transforms.Transform

Abstraction of a batch transform

abstract __call__(self, batch)

class trw.transforms.transforms.TransformBatchWithCriteria(criteria_fn, transform_fn)

Bases: Transform

Helper function to apply a given transform function on features that satisfy a criteria

__call__(self, batch)

class trw.transforms.transforms.TransformBatchJointWithCriteria(criteria_fn, transform_fn)

Bases: Transform

Helper function to apply a given transform function on features that satisfy a criteria. Feature will be jointly transformed

__call__(self, batch)

trw.transforms.transforms.criteria_is_array_3_or_above(feature_name, feature_value)

Return True if the feature is a numpy or torch array dim >= 3

trw.transforms.transforms.criteria_feature_name(feature_name, feature_value, feature_names)

Return True if the feature name belongs to a given set of names