Compocyte.core.models.trees¶

CatBoost-based boosted trees wrapper.

This module wraps CatBoost models used as one of the available local classifier options. It provides training/prediction wrappers and load/save helpers that the hierarchical classifier expects.

class Compocyte.core.models.trees.BoostedTrees(labels=None, labels_enc=None, labels_dec=None, model=None, catboost_kwargs={}, **kwargs)[source]¶