Augmentation¶
- class metatrain.utils.augmentation.RotationalAugmenter(target_info_dict: Dict[str, TargetInfo], extra_data_info_dict: Dict[str, TargetInfo] | None = None)[source]¶
Bases:
object
A class to apply random rotations and inversions to a set of systems and their targets.
- Parameters:
- apply_random_augmentations(systems: List[System], targets: Dict[str, TensorMap], extra_data: Dict[str, TensorMap] | None = None) Tuple[List[System], Dict[str, TensorMap], Dict[str, TensorMap]] [source]¶
Apply a random augmentation to a number of
System
objects and its targets.