diff --git a/modules/ml/src/tree.cpp b/modules/ml/src/tree.cpp index 4faf9b3b5f..6a596551ea 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; } }