ForeTiS.model._baseline_model

Module Contents

Classes

BaselineModel

Parent class based on BaseModel for all baseline models to share functionalities
class ForeTiS.model._baseline_model.BaselineModel(optuna_trial, datasets, featureset_name, pca_transform, target_column, optimize_featureset)[source]

Bases: ForeTiS.model._base_model.BaseModel, abc.ABC

Parent class based on BaseModel for all baseline models to share functionalities See BaseModel for more information.

Parameters:

  • optuna_trial (optuna.trial.Trial) –

  • datasets (list) –

  • featureset_name (str) –

  • pca_transform (bool) –

  • target_column (str) –

  • optimize_featureset (bool) –

retrain(retrain)[source]

Implementation of the retraining for baseline models. See BaseModel for more information.

Parameters:

retrain (pandas.DataFrame) –

update(update, period)[source]

Implementation of the retraining for baseline models. See BaseModel for more information.

Parameters:

  • update (pandas.DataFrame) –

  • period (int) –

predict(X_in)[source]

Implementation of a prediction based on input features for baseline models. See BaseModel for more information.

Parameters:

X_in (pandas.DataFrame) –

Return type:

numpy.array

train_val_loop(train, val)[source]

Implementation of a train and validation loop for baseline models. See BaseModel for more information.

Parameters:

  • train (pandas.DataFrame) –

  • val (pandas.DataFrame) –

Return type:

numpy.array