tfdf.py_tree.tree.Tree

A single decision tree.

tfdf.py_tree.tree.Tree(
    root: Optional[tfdf.py_tree.node.AbstractNode],
    label_classes: Optional[List[str]] = None
)

Attributes
label_classes
root

Methods

pretty

View source

pretty(
    max_depth: Optional[int] = 4
) -> str

Returns a readable textual representation of the tree.

Unlike repr(tree), tree.pretty() format the representation (line return, margin, hide class names) to improve readability.

This representation can be changed and codes should not try to parse the output of pretty. To access programmatically the tree structure, use root().

Args
max_depth	The maximum depth of the nodes to display. Deeper nodes are skipped and replaced by "...". If not specified, prints the entire tree.

Returns
A pretty-string representing the tree.