mmc.plot_tree()
-
mmochi.plotting.plot_tree(hierarchy, level, tree_number=random.randint(1, 10), save='tree.png') Plots a tree from the random forest at a specified level of the classifier. Note these plots are often very unwieldy. See export_graphviz for details.
requires Graphviz which can be obtained with sudo apt-get install graphviz
Parameters: - hierarchy – Hierarchy object which random forest is plotting on
- level (
str) – Name of the level of interest to pull the tree, must have a trained classifier attached to it - tree_number (
int(default:random.randint(1,10))) – The tree to pick. If undefined, a random tree. - save (
str(default:'tree.png')) – The name to save the plot as a .png format. This must be done before displaying the plot as well