blocksnet.machine_learning.strategy.catboost

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

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

Bases: ClassificationBase, CatBoostBaseStrategy

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

class blocksnet.machine_learning.strategy.catboost.CatBoostRegressionStrategy(model_params: dict | None = None)[source]: Bases: CatBoostBaseStrategy

Modules

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