Hiding threshold if equals to zero in tree representation

cavalab · Oct 29, 2024 · 10a4caa · 10a4caa
1 parent 82995a2
commit 10a4caa
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/program/tree_node.cpp b/src/program/tree_node.cpp
@@ -39,6 +39,9 @@ string TreeNode::get_tree_model(bool pretty, string offset) const
     }
 
     if (Is<NodeType::SplitBest>(data.node_type)){
+        if (data.W == 0.0) // likely it is a boolean (threshold is forced at zero)
+            return fmt::format("If({})", data.get_feature()) + child_outputs;
+
         return fmt::format("If({}>{:.2f})", data.get_feature(), data.W) +
                child_outputs;
     }