saealib.TrainingData#

class saealib.TrainingData(train_x, train_y)[source]#

Bases: object

Immutable container for the training data passed to Surrogate.fit.

Parameters:

  • train_x (ndarray)

  • train_y (ndarray)

train_x#

Design variable matrix. shape: (n_train, dim). For PairwiseComparisonSet the shape is (n_train, 2 * dim).

Type:

np.ndarray

train_y#

Target values. shape: (n_train, n_obj) for regression, (n_train,) for classification or ranking.

Type:

np.ndarray

Methods

__init__

Method Details

TrainingData.__init__(train_x, train_y)#
Parameters:

  • train_x (ndarray)

  • train_y (ndarray)

Return type:

None