learner

XGBoostClassifierLearner(**kwargs)

Bases: SupervisedLearner

Wrapper for XGBoost classifiers.

Source code in src/flowcean/xgboost/learner.py

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

def __init__(self, **kwargs: Any) -> None:
    self.regressor = XGBRegressor(
        **kwargs,
    )
    super().__init__()