blocksnet.machine_learning.strategy

class blocksnet.machine_learning.strategy.BaseStrategy(model_cls, model_params: dict = {})[source]

Bases: ABC

property model
abstract train(*args, **kwargs)[source]
abstract predict(*args, **kwargs) ndarray[source]
save(path: str)[source]

Must call super().save(path) if overridden.

load(path: str)[source]

Must call super().load(path) if overridden.

class blocksnet.machine_learning.strategy.ClassificationBase[source]

Bases: ABC

abstract predict_proba(*args, **kwargs)[source]

Modules

blocksnet.machine_learning.strategy.base_strategy

blocksnet.machine_learning.strategy.catboost

blocksnet.machine_learning.strategy.classification_base

blocksnet.machine_learning.strategy.sklearn

blocksnet.machine_learning.strategy.xgboost