SurrogateProtocol

class baybe.surrogates.base.SurrogateProtocol

Bases: Protocol

Type protocol specifying the interface surrogate models need to implement.

Public methods

__init__(*args, **kwargs)
fit(searchspace, objective, measurements)	Fit the surrogate to training data in a given modelling context.
to_botorch()	Create the botorch-ready representation of the fitted model.

__init__(*args, **kwargs)

fit(searchspace: SearchSpace, objective: Objective, measurements: DataFrame)

Fit the surrogate to training data in a given modelling context.

For details on the expected method arguments, see baybe.recommenders.base.RecommenderProtocol().

Return type:

None

to_botorch()

Create the botorch-ready representation of the fitted model.

The botorch.models.model.Model created by this method needs to be configured such that it can be called with candidate points in computational representation, that is, input of the form as obtained via baybe.searchspace.core.SearchSpace.transform().

Return type:

Model