lightgbm.create_tree_digraph

lightgbm.create_tree_digraph(booster, tree_index=0, show_info=None, precision=3, orientation='horizontal', **kwargs)[source]

Create a digraph representation of specified tree.

Each node in the graph represents a node in the tree.

Non-leaf nodes have labels like Column_10 <= 875.9, which means “this node splits on the feature named “Column_10”, with threshold 875.9”.

Leaf nodes have labels like leaf 2: 0.422, which means “this node is a leaf node, and the predicted value for records that fall into this node is 0.422”. The number (2) is an internal unique identifier and doesn’t have any special meaning.

Note

For more information please visit https://graphviz.readthedocs.io/en/stable/api.html#digraph.

Parameters
  • booster (Booster or LGBMModel) – Booster or LGBMModel instance to be converted.

  • tree_index (int, optional (default=0)) – The index of a target tree to convert.

  • show_info (list of str, or None, optional (default=None)) –

    What information should be shown in nodes.

    • 'split_gain' : gain from adding this split to the model

    • 'internal_value' : raw predicted value that would be produced by this node if it was a leaf node

    • 'internal_count' : number of records from the training data that fall into this non-leaf node

    • 'internal_weight' : total weight of all nodes that fall into this non-leaf node

    • 'leaf_count' : number of records from the training data that fall into this leaf node

    • 'leaf_weight' : total weight (sum of Hessian) of all observations that fall into this leaf node

    • 'data_percentage' : percentage of training data that fall into this node

  • precision (int or None, optional (default=3)) – Used to restrict the display of floating point values to a certain precision.

  • orientation (str, optional (default='horizontal')) – Orientation of the tree. Can be ‘horizontal’ or ‘vertical’.

  • **kwargs – Other parameters passed to Digraph constructor. Check https://graphviz.readthedocs.io/en/stable/api.html#digraph for the full list of supported parameters.

Returns

graph – The digraph representation of specified tree.

Return type

graphviz.Digraph