caikit.core.augmentors.base
Attributes
Classes
Module Contents
- caikit.core.augmentors.base.error
- class caikit.core.augmentors.base.AugmentorBase(random_seed, produces_none=False)[source]
- produces_none = False
- augment(inp_obj)[source]
Take an object in, give an object back. Calls ._augment in the subclass.
- Args:
- inp_obj (str | caikit.core.data_model.DataBase): Object to be
augmented.
- Returns:
- str | caikit.core.data_model.DataBase: Augmented object of same type
as input inp_obj.
- reset()[source]
Reset random number generation for the current augmentor. Note that this currently assumes the augmentor is using the builtin random generator leveraged by Python; if you end up using something else, you may want to override this or restructure this base class to allow resetting of random states based on seed type.