View source on GitHub
|
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
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. |