(self,
preprocessing: str,
exp_type: str,
projector_hidden_dim: int = 128, # only if preprocessing == 'mlp'
projector_n_layers: int = 1, # only if preprocessing == 'mlp'
projector_net_normalization: str = 'layer_norm', # only if preprocessing == 'mlp'
use_level_features: bool = True,
)
| 359 | ONLY_LEVEL_NODES = ['graph_net_level_nodes'] |
| 360 | |
| 361 | def __init__(self, |
| 362 | preprocessing: str, |
| 363 | exp_type: str, |
| 364 | projector_hidden_dim: int = 128, # only if preprocessing == 'mlp' |
| 365 | projector_n_layers: int = 1, # only if preprocessing == 'mlp' |
| 366 | projector_net_normalization: str = 'layer_norm', # only if preprocessing == 'mlp' |
| 367 | use_level_features: bool = True, |
| 368 | ): |
| 369 | self.preprocessing_type = preprocessing.lower() |
| 370 | self.projector_hidden_dim = projector_hidden_dim |
| 371 | self.projector_n_layers = projector_n_layers |
| 372 | self.projector_net_normalization = projector_net_normalization |
| 373 | self.use_level_features = use_level_features |
| 374 | |
| 375 | def get_preprocesser(self, batched: bool = False, verbose: bool = True): |
| 376 | if self.preprocessing_type in ['mlp', 'mlp_projection']: |
nothing calls this directly
no outgoing calls
no test coverage detected