New! Use Simple ML for Sheets to apply machine learning to the data in your Google Sheets Read More

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
)

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.