| 186 | |
| 187 | |
| 188 | class AllAreNodesTransforms(AbstractGraphTransform, ABC): |
| 189 | def __init__(self, exp_type: str, use_level_features: bool = True): |
| 190 | super().__init__(exp_type) |
| 191 | self.use_level_features = use_level_features |
| 192 | if self.use_level_features: |
| 193 | self._n_nodes = self.spatial_input_dim[LAYERS] + self.spatial_input_dim[LEVELS] + 1 |
| 194 | self.LEVEL_NODES = slice(1, self.n_nodes, 2) # start at 1, then every second, [1, 3, 5, 7,...] |
| 195 | self.LAYER_NODES = slice(2, self.n_nodes, 2) # [2, 4, 6,...] |
| 196 | else: |
| 197 | self._n_nodes = self.spatial_input_dim[LAYERS] + 1 |
| 198 | self.LAYER_NODES = slice(1, self.n_nodes) # [1, 2, 3,...] self.padding_value = padding_value |
| 199 | self.GLOBAL_NODE = 0 |
| 200 | |
| 201 | |
| 202 | class EdgesAndNodesTransforms(AbstractGraphTransform, ABC): |
nothing calls this directly
no outgoing calls
no test coverage detected