blocksnet.machine_learning.strategy

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

Bases: ABC

property model

abstract train(*args, **kwargs)

abstract predict(*args, **kwargs) → ndarray

save(path: str)

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

load(path: str)

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

class blocksnet.machine_learning.strategy.ClassificationBase

Bases: ABC

abstract predict_proba(*args, **kwargs)

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