xgboost

is_xgboost_version_at_least(major: int, minor: Optional[int] = None, patch: Optional[int] = None)

class XGBGradientBoostedVectorRegressionModel(random_state=42, **model_args)

Bases: sensai.sklearn.sklearn_base.AbstractSkLearnMultipleOneDimVectorRegressionModel, sensai.sklearn.sklearn_base.FeatureImportanceProviderSkLearnRegressionMultipleOneDim

XGBoost’s regression model using gradient boosted trees

__init__(random_state=42, **model_args)

Parameters

model_args – See https://xgboost.readthedocs.io/en/latest/python/python_api.html#xgboost.XGBRegressor

class XGBRandomForestVectorRegressionModel(random_state=42, **model_args)

Bases: sensai.sklearn.sklearn_base.AbstractSkLearnMultipleOneDimVectorRegressionModel, sensai.sklearn.sklearn_base.FeatureImportanceProviderSkLearnRegressionMultipleOneDim

XGBoost’s random forest regression model

__init__(random_state=42, **model_args)

Parameters

model_args – See https://xgboost.readthedocs.io/en/latest/python/python_api.html#xgboost.XGBRFRegressor

class XGBGradientBoostedVectorClassificationModel(random_state=42, use_balanced_class_weights=False, **model_args)

Bases: sensai.sklearn.sklearn_base.AbstractSkLearnVectorClassificationModel, sensai.sklearn.sklearn_base.FeatureImportanceProviderSkLearnClassification

XGBoost’s classification model using gradient boosted trees

__init__(random_state=42, use_balanced_class_weights=False, **model_args)

Parameters

model_args – See https://xgboost.readthedocs.io/en/latest/python/python_api.html#xgboost.XGBClassifier

class XGBRandomForestVectorClassificationModel(random_state=42, use_balanced_class_weights=False, **model_args)

Bases: sensai.sklearn.sklearn_base.AbstractSkLearnVectorClassificationModel, sensai.sklearn.sklearn_base.FeatureImportanceProviderSkLearnClassification

XGBoost’s random forest classification model

__init__(random_state=42, use_balanced_class_weights=False, **model_args)

Parameters

model_args – See https://xgboost.readthedocs.io/en/latest/python/python_api.html#xgboost.XGBRFClassifier