deepcave.evaluators.epm.random_forest_surrogate

# RandomForest Surrogate

This module provides a RandomForest Surrogate model.

Mean and standard deviation values can be predicted for a given input with this module.

## Classes
  • RandomForestSurrogate: Random forest surrogate for the pyPDP package.

Classes

RandomForestSurrogate(configspace[, seed, ...])

Random forest surrogate for the pyPDP package.

class deepcave.evaluators.epm.random_forest_surrogate.RandomForestSurrogate(configspace, seed=None, n_trees=16)[source]

Bases: SurrogateModel

Random forest surrogate for the pyPDP package.

Predict deviations and fit the model.

predict(X)[source]

Predict the deviations.

Parameters:

X (np.ndarray) – The data points on which to predict.

Returns:

The means and standard deviation.

Return type:

Tuple[np.ndarray, np.ndarray]