10000 fix: incorrect type cast in median array generation for node_impurity · scikit-learn/scikit-learn@b53c1c5 · GitHub
[go: up one dir, main page]

Skip to content

Commit b53c1c5

Browse files
committed
fix: incorrect type cast in median array generation for node_impurity
1 parent 7d97c45 commit b53c1c5

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

sklearn/tree/_criterion.pyx

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -974,7 +974,7 @@ cdef class MAE(RegressionCriterion):
974974
"""Evaluate the impurity of the current node, i.e. the impurity of
975975
samples[start:end]"""
976976
cdef double* medians = NULL
977-
medians = <DOUBLE_t *> calloc(self.n_outputs, sizeof(double))
977+
medians = <double*> calloc(self.n_outputs, sizeof(double))
978978
if (medians == NULL):
979979
with gil:
980980
raise MemoryError()

0 commit comments

Comments
 (0)
0