Fix Invalid decision type error while plotting tree in Python (#810)

* Fix Invalid decision type error while plotting tree in Python * Fix Invalid decision type error while plotting tree in Python (modified = to ==)
microsoft · Aug 18, 2017 · 8a2b644 · 8a2b644
1 parent 9d3226e
commit 8a2b644
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python-package/lightgbm/plotting.py b/python-package/lightgbm/plotting.py
@@ -268,9 +268,9 @@ def add(root, parent=None, decision=None):
                 if info in {'split_gain', 'internal_value', 'internal_count'}:
                     label += '\n' + info + ':' + str(root[info])
             graph.node(name, label=label)
-            if root['decision_type'] == 'no_greater':
+            if root['decision_type'] == '<=':
                 l_dec, r_dec = '<=', '>'
-            elif root['decision_type'] == 'is':
+            elif root['decision_type'] == '==':
                 l_dec, r_dec = 'is', "isn't"
             else:
                 raise ValueError('Invalid decision type in tree model.')