odtlearn.utils.TreePlotter#

Module Contents#

Classes#

MPLPlotter

class odtlearn.utils.TreePlotter.MPLPlotter(tree, node_dict, column_names, max_depth, classes, model_name, label='all', filled=False, rounded=False, precision=3, fontsize=None, color_dict={'node': None, 'leaves': []}, edge_annotation=True, arrow_annotation_font_scale=0.5, debug=False)[source]#

Bases: sklearn.tree._export._MPLTreeExporter

get_fill_color(node_id)[source]#
node_to_str(node_id, leaf, selected_feature, cutoff, value)[source]#
export(ax=None, distance=1.0)[source]#
recurse(node, ax, max_x, max_y, depth=0)[source]#