syne_tune.optimizer.schedulers.searchers.conformal.surrogate.quantile_regression_surrogate module

class syne_tune.optimizer.schedulers.searchers.conformal.surrogate.quantile_regression_surrogate.QuantileRegressionSurrogateModel(config_space, mode, random_state=None, max_fit_samples=None, quantiles=5, valid_fraction=0.0, min_samples_to_conformalize=None, **kwargs)[source]

Bases: SurrogateModel

predict(df_features)[source]

This will need quantiles, median, mean

Return type:

QuantileRegressorPredictions