ForeTiS.model.xgboost

Module Contents

Classes

XgBoost

Implementation of a class for XGBoost.

class ForeTiS.model.xgboost.XgBoost(optuna_trial, datasets, featureset_name, pca_transform, target_column, optimize_featureset)

Bases: ForeTiS.model._sklearn_model.SklearnModel

Implementation of a class for XGBoost.

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:

xgboost.XGBModel

define_hyperparams_to_tune()

See BaseModel for more information on the format.

Return type:

dict