blocksnet.machine_learning.strategy.xgboost.base_strategy

Classes

XGBoostBaseStrategy(model_cls[, model_params])

class blocksnet.machine_learning.strategy.xgboost.base_strategy.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]