| """Augmentation strategies for GraPHFormer.""" |
|
|
| from .tree_augmentations import ( |
| Compose, |
| RandomDropSubTrees, RandomSkipParentNode, RandomSwapSiblingSubTrees, |
| RandomRotate, RandomJitter, RandomShift, RandomFlip, |
| RandomScaleCoords, RandomScaleFeats, RandomMaskFeats, |
| RandomElasticate, RandomJitterLength, |
| ) |
| from .persistence_augmentations import ( |
| PersistenceSpaceAugmentation, SigmaVariationAugmentation, |
| CombinedPersistenceAugmentation, get_default_augmentation, |
| ) |
|
|
| __all__ = [ |
| "Compose", |
| "RandomDropSubTrees", "RandomSkipParentNode", "RandomSwapSiblingSubTrees", |
| "RandomRotate", "RandomJitter", "RandomShift", "RandomFlip", |
| "RandomScaleCoords", "RandomScaleFeats", "RandomMaskFeats", |
| "RandomElasticate", "RandomJitterLength", |
| "PersistenceSpaceAugmentation", "SigmaVariationAugmentation", |
| "CombinedPersistenceAugmentation", "get_default_augmentation", |
| ] |
|
|
