dgld.modules.dglAug.augs

Graph Augmentation Adapted from https://github.com/PyGCL/PyGCL/blob/main/GCL/augmentors/augmentor.py

class dgld.modules.dglAug.augs.ComposeAug(augs: List[BaseTransform], cross: bool = True)[source]

Bases: BaseTransform

Execute graph augments in sequence.

Parameters

  • augs (List[BaseTransform]) – graphs augments using DGL tansform

  • cross (bool, optional) – if use cross graph augments, by default True

class dgld.modules.dglAug.augs.RandomChoiceAug(augs: List[BaseTransform], n_choices: int, cross: bool = True)[source]

Bases: BaseTransform

Execute graph augments in random.

Parameters

  • augs (_type_) – _description_

  • n_choices (_type_) – _description_

  • cross (bool, optional) – _description_, by default True