|
- from .views_fn import (
- DropNode,
- PermuteEdge,
- MaskNode,
- SubGraph,
- RandomView
- )
-
- def get_view_by_name(view, aug_ratio):
- if view is None:
- return lambda x: x
- elif view == "dropN":
- return DropNode(aug_ratio=aug_ratio)
- elif view == "permE":
- return PermuteEdge(aug_ratio=aug_ratio)
- elif view == "subgraph":
- return SubGraph(aug_ratio=aug_ratio)
- elif view == "maskN":
- return MaskNode(aug_ratio=aug_ratio)
- elif view == "random2":
- canditates = [DropNode(aug_ratio=aug_ratio),
- SubGraph(aug_ratio=aug_ratio)]
- return RandomView(candidates=canditates)
- elif view == "random3":
- canditates = [DropNode(aug_ratio=aug_ratio),
- SubGraph(aug_ratio=aug_ratio),
- PermuteEdge(aug_ratio=aug_ratio)]
- return RandomView(candidates=canditates)
- elif view == "random4":
- canditates = [DropNode(aug_ratio=aug_ratio),
- SubGraph(aug_ratio=aug_ratio),
- PermuteEdge(aug_ratio=aug_ratio),
- MaskNode(aug_ratio=aug_ratio)]
- return RandomView(candidates=canditates)
- else:
- raise NotImplementedError(f'{view} is not supported yet. Support: ["dropN", "permE", "subgraph", \
- "maskN", "random2", "random3", "random4", None]')
|
