vertebrae.extractors.tree_leaf

Leaf-index embedding extractor for tree ensembles.

Classes

TreeLeafEmbeddingExtractor

Extract dense or sparse leaf embeddings from fitted tree ensembles.

Module Contents

class vertebrae.extractors.tree_leaf.TreeLeafEmbeddingExtractor(name, model, backend='auto', encoding='dense', sparse_output=True, recipe_data=None, cache_identity=None)

Extract dense or sparse leaf embeddings from fitted tree ensembles.

Parameters:

• name (str)

• model (Any)

• backend (str)

• encoding (str)

• sparse_output (bool)

• recipe_data (Optional[Dict[str, Any]])

• cache_identity (Optional[str])