Skip to content

GitLab

Commit 2e6a97fb authored by Tsukasa OMOTO's avatar Tsukasa OMOTO Committed by Guolin Ke
Browse files

python-package: plot leaf index (#457)

parent 001fe536
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 2 deletions
python-package/lightgbm/plotting.py
View file @ 2e6a97fb
......@@ -277,8 +277,9 @@ def _to_graphviz(tree_info, show_info, feature_names,
add(root['left_child'], name, l_dec)
add(root['right_child'], name, r_dec)
else: # leaf
name = 'left' + str(root['leaf_index'])
label = 'leaf_value:' + str(root['leaf_value'])
name = 'leaf' + str(root['leaf_index'])
label = 'leaf_index:' + str(root['leaf_index'])
label += '\nleaf_value:' + str(root['leaf_value'])
if 'leaf_count' in show_info:
label += '\nleaf_count:' + str(root['leaf_count'])
graph.node(name, label=label)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment