gpytorchwrapper.src.config.model_factory

Functions

`get_likelihood`(likelihood_conf)	Get the likelihood class and options
`get_model`(model_conf)	Get the model class and options
`get_optimizer`(optimizer_conf)	Get the optimizer class and options
`get_plugins`([path])
`get_transformer`(transformer_conf)	Get the transformer class and options

gpytorchwrapper.src.config.model_factory.get_likelihood(likelihood_conf: LikelihoodConf) → Likelihood[source]

Get the likelihood class and options

Parameters:: likelihood_conf (dict) – Dictionary containing the likelihood specifications
Returns:: selected_likelihood_class – The selected likelihood class
Return type:: object

gpytorchwrapper.src.config.model_factory.get_model(model_conf: ModelConf) → ExactGP[source]

Get the model class and options

Parameters:: model_conf (ModelConf) – dataclass containing the model specifications
Returns:: selected_model_class – The selected model class
Return type:: object

gpytorchwrapper.src.config.model_factory.get_optimizer(optimizer_conf: OptimizerConf) → Optimizer[source]

Get the optimizer class and options

Parameters:: optimizer_conf (OptimizerConf) – dataclass containing the optimizer specifications
Returns:: selected_optimizer_class – The selected optimizer class
Return type:: object

gpytorchwrapper.src.config.model_factory.get_plugins(path: str | None = None) → dict[str, ModuleType][source]

Parameters:: path (str or None, optional) – path to the directory containing the model plugins
Returns:: discovered_plugins – dict with the names of the model class as a string and the model classes as values
Return type:: dict

gpytorchwrapper.src.config.model_factory.get_transformer(transformer_conf: TransformerConf) → object[source]

Get the transformer class and options

Parameters:: transformer_conf (TransformerConf) – dataclass containing the transformer specifications
Returns:: selected_transformer_class – The selected scikit-learn transformer class
Return type:: object