GOOD.kernel.pipeline_manager

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

Functions

load_pipeline(name, task, model, loader, ...)

A pipeline loader.

GOOD.kernel.pipeline_manager.load_pipeline(name: str, task: str, model: Module, loader: Union[DataLoader, Dict[str, DataLoader]], ood_algorithm: BaseOODAlg, config: Union[CommonArgs, Munch]) → Pipeline

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

Returns

A instantiated pipeline.