Merge pull request #11171 from codingforfun:fix_11143

7 years ago · 284e5231c5
parent 6b041c8aeb 24e2e0d3f9
commit 284e5231c5
1 changed files with 1 additions and 0 deletions
--- a/modules/ml/src/tree.cpp
+++ b/modules/ml/src/tree.cpp
@ -632,6 +632,7 @@ void DTreesImpl::calcValue( int nidx, const vector<int>& _sidx )
        }

        node->node_risk = sum2 - (sum/sumw)*sum;
+        node->node_risk /= sumw;
        node->value = sum/sumw;
    }
 }