From 044c2a4fbdd05afa38f0efa364b5c8fdf9fbe879 Mon Sep 17 00:00:00 2001 From: Marcus Edel Date: Mon, 12 Apr 2021 03:48:53 +0200 Subject: [PATCH] Catch potential division by zero. --- src/mlpack/core/tree/rectangle_tree/x_tree_split_impl.hpp | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/mlpack/core/tree/rectangle_tree/x_tree_split_impl.hpp b/src/mlpack/core/tree/rectangle_tree/x_tree_split_impl.hpp index c977f0fecb..cc1119a845 100644 --- a/src/mlpack/core/tree/rectangle_tree/x_tree_split_impl.hpp +++ b/src/mlpack/core/tree/rectangle_tree/x_tree_split_impl.hpp @@ -435,7 +435,8 @@ bool XTreeSplit::SplitNonLeafNode(TreeType *tree, std::vector& relevels) bool useMinOverlapSplit = false; if (tiedOnOverlap) { - if (overlapBestAreaAxis / areaBestAreaAxis < MAX_OVERLAP) + if (MAX_OVERLAP > 0 && + overlapBestAreaAxis / areaBestAreaAxis < MAX_OVERLAP) { tree->numDescendants = 0; tree->bound.Clear();