diff --git a/modules/ml/src/tree.cpp b/modules/ml/src/tree.cpp index c02b7440e7..4b3079b3d1 100644 --- a/modules/ml/src/tree.cpp +++ b/modules/ml/src/tree.cpp @@ -632,6 +632,7 @@ void DTreesImpl::calcValue( int nidx, const vector& _sidx ) } node->node_risk = sum2 - (sum/sumw)*sum; + node->node_risk /= sumw; node->value = sum/sumw; } }