GOOD.networks.model_manager

A module that is consist of a GNN model loader and model configuration function.

Functions

load_model(name, config)

A model loader.

GOOD.networks.model_manager.load_model(name: str, config: Union[CommonArgs, Munch]) → Module

A model loader. :param name: Name of the chosen GNN. :type name: str :param config: Please refer to specific GNNs for required configs and formats. :type config: Union[CommonArgs, Munch]

Returns

A instantiated GNN model.