ForeTiS.model._additionalmodels.gpr_sklearn

Module Contents

Classes

Gpr

Implementation of a class for Gpr.
class ForeTiS.model._additionalmodels.gpr_sklearn.Gpr(optuna_trial, datasets, featureset_name, pca_transform, target_column, optimize_featureset)

Bases: ForeTiS.model._sklearn_model.SklearnModel

Implementation of a class for Gpr.

See BaseModel for more information on the attributes.

Parameters:

  • optuna_trial (optuna.trial.Trial) –

  • datasets (list) –

  • featureset_name (str) –

  • pca_transform (bool) –

  • target_column (str) –

  • optimize_featureset (bool) –

define_model()

Definition of the actual prediction model.

See BaseModel for more information.

Return type:

sklearn.gaussian_process.GaussianProcessRegressor

define_hyperparams_to_tune()

See BaseModel for more information on the format.

Return type:

dict

extend_kernel_combinations()

Function extending kernels list with combinations based on base_kernels