Make sure its defined

Signed-off-by: Adam Li <adam2392@gmail.com>
neurodata · Jul 11, 2024 · ae2604b · ae2604b
1 parent 05f2f9f
commit ae2604b
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/sklearn/tree/_splitter.pyx b/sklearn/tree/_splitter.pyx
@@ -959,6 +959,13 @@ cdef inline int node_split_random(
             current_split.threshold
         )
 
+        if missing_go_to_left:
+            n_left = current_split.pos - start + n_missing
+            n_right = end_non_missing - current_split.pos
+        else:
+            n_left = current_split.pos - start
+            n_right = end_non_missing - current_split.pos + n_missing
+
         # Reject if min_samples_leaf is not guaranteed
         if splitter.check_presplit_conditions(&current_split, n_missing, missing_go_to_left) == 1:
             continue