8000
We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent de968ed commit 6d173c0Copy full SHA for 6d173c0
sklearn/ensemble/_hist_gradient_boosting/splitting.pyx
@@ -569,15 +569,15 @@ cdef class Splitter:
569
free(split_infos)
570
return out
571
572
- cdef unsigned int _find_best_feature_to_split_helper(
+ cdef int _find_best_feature_to_split_helper(
573
self,
574
split_info_struct * split_infos, # IN
575
int n_allowed_features,
576
) noexcept nogil:
577
"""Return the index of split_infos with the best feature split."""
578
cdef:
579
- unsigned int split_info_idx
580
- unsigned int best_split_info_idx = 0
+ int split_info_idx
+ int best_split_info_idx = 0
581
582
for split_info_idx in range(1, n_allowed_features):
583
if (split_infos[split_info_idx].gain > split_infos[best_split_info_idx].gain):
0 commit comments