baybe.surrogates.utilsΒΆ

Utilities for working with surrogates.

Functions

batchify(posterior)

Wrap Surrogate posterior functions to enable proper batching.

catch_constant_targets(model_cls)

Wrap a Surrogate class that cannot handle constant training target values.

scale_model(model_cls)

Wrap a Surrogate class such that it operates with scaled representations.