blocksnet.machine_learning.strategy.xgboost

class blocksnet.machine_learning.strategy.xgboost.XGBoostBaseStrategy(model_cls: type[xgboost.sklearn.XGBModel], model_params: dict | None = None)[source]

Bases: BaseStrategy, ABC

train(x_train: ndarray, x_test: ndarray, y_train: ndarray, y_test: ndarray) → float[source]

predict(x: ndarray) → ndarray[source]

class blocksnet.machine_learning.strategy.xgboost.XGBoostClassificationStrategy(model_params: dict | None = None)[source]

Bases: ClassificationBase, XGBoostBaseStrategy

predict_proba(x: ndarray) → ndarray[source]

class blocksnet.machine_learning.strategy.xgboost.XGBoostRegressionStrategy(model_params: dict | None = None)[source]: Bases: XGBoostBaseStrategy

Modules

`blocksnet.machine_learning.strategy.xgboost.base_strategy`
`blocksnet.machine_learning.strategy.xgboost.classification_strategy`
`blocksnet.machine_learning.strategy.xgboost.regression_strategy`