| 287 | self.padding_value = padding_value |
| 288 | |
| 289 | def transform(self, X: Dict[str, np.ndarray]) -> np.ndarray: |
| 290 | global_node, levels, layers = X[GLOBALS], X[LEVELS], X[LAYERS] |
| 291 | # set all values to padding by default |
| 292 | padded_data = self.padding_value * np.ones((self.n_nodes, self.output_dim)) |
| 293 | padded_data[self.GLOBAL_NODE, :self.input_dim[GLOBALS]] = global_node |
| 294 | padded_data[self.LAYER_NODES, :self.input_dim[LAYERS]] = layers |
| 295 | if self.use_level_features: |
| 296 | padded_data[self.LEVEL_NODES, :self.input_dim[LEVELS]] = levels |
| 297 | return padded_data |
| 298 | |
| 299 | def batched_transform(self, X: Dict[str, np.ndarray]) -> np.ndarray: |
| 300 | global_node, levels, layers = X[GLOBALS], X[LEVELS], X[LAYERS] |