Export shift_missing_values_to_left_if_required

Signed-off-by: Adam Li <adam2392@gmail.com>
neurodata · Mar 10, 2024 · b61ae3d · b61ae3d
1 parent 5ccd00f
commit b61ae3d
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/sklearn/tree/_splitter.pxd b/sklearn/tree/_splitter.pxd
@@ -140,3 +140,9 @@ cdef class Splitter(BaseSplitter):
         float64_t lower_bound,
         float64_t upper_bound
     ) noexcept nogil
+
+cdef void shift_missing_values_to_left_if_required(
+    SplitRecord* best,
+    intp_t[::1] samples,
+    intp_t end,
+) noexcept nogil