baybe.surrogates.utils

Utilities for working with surrogates.

Functions

batchify_ensemble_predictor(base_predictor)	Wrap an ensemble predictor to make it evaluate t-batches as an augmented q-batch.
batchify_mean_var_prediction(posterior)	Wrap a posterior method to make it evaluate t-batches as an augmented q-batch.
catch_constant_targets(cls[, std_threshold])	Make a Surrogate class robustly handle constant training targets.