Skip to content

learner

XGBoostClassifierLearner(**kwargs)

Bases: SupervisedLearner

Wrapper for XGBoost classifiers.

Source code in src/flowcean/xgboost/learner.py
14
15
16
def __init__(self, **kwargs: Any) -> None:
    self.classifier = XGBClassifier(**kwargs)
    super().__init__()

XGBoostRegressorLearner(**kwargs)

Bases: SupervisedLearner

Wrapper for XGBoost regressor.

Source code in src/flowcean/xgboost/learner.py
44
45
46
def __init__(self, **kwargs: Any) -> None:
    self.regressor = XGBRegressor(**kwargs)
    super().__init__()