Have a question? Connect with the community at the TensorFlow Forum Visit Forum

tfdf.model_plotter.plot_tree

Plots a decision tree.

tree A decision tree.
max_depth Maximum plotting depth. Makes the plot more readable in case of large trees.

The html content displaying the tree.