Fix.

dmlc · Jun 11, 2021 · 46afb0d · 46afb0d
1 parent 326aecb
commit 46afb0d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/tree/tree_model.cc b/src/tree/tree_model.cc
@@ -648,8 +648,8 @@ class GraphvizGenerator : public TreeGenerator {
         {"{cond}",   has_less ? SuperT::ToStr(cond) : ""},
         {"{params}", param_.condition_node_params}});
 
-    result += BuildEdge<true>(tree, nid, tree[nid].LeftChild(), true);
-    result += BuildEdge<true>(tree, nid, tree[nid].RightChild(), false);
+    result += BuildEdge<false>(tree, nid, tree[nid].LeftChild(), true);
+    result += BuildEdge<false>(tree, nid, tree[nid].RightChild(), false);
 
     return result;
   };