blocksnet.machine_learning.strategy.catboost.base_strategy

Classes

CatBoostBaseStrategy(model_cls[, model_params])

class blocksnet.machine_learning.strategy.catboost.base_strategy.CatBoostBaseStrategy(model_cls: type[catboost.core.CatBoost], 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]