saealib.Surrogate

class saealib.Surrogate[source]

Bases: ABC

Base class for surrogate models.

Methods

__init__

fit

Fit the surrogate model.

predict

Predict using the surrogate model.

Method Details

Surrogate.__init__()
abstract Surrogate.fit(train_x, train_y)[source]

Fit the surrogate model.

Parameters:

  • train_x (np.ndarray) – Training input data. shape: (n_samples, n_features)

  • train_y (np.ndarray) – Training output data. shape: (n_samples, n_obj) For n_obj == 1, shape (n_samples,) is also accepted.

Return type:

None

abstract Surrogate.predict(test_x)[source]

Predict using the surrogate model.

Parameters:

test_x (np.ndarray) – Input data for prediction. shape: (n_samples, n_features)

Returns:

prediction.value shape: (n_samples, n_obj) prediction.std shape: (n_samples, n_obj) or None

Return type:

SurrogatePrediction