Merge pull request #11171 from codingforfun:fix_11143

284e5231 · Vadim Pisarevsky · 6b041c8a · 24e2e0d3 · 284e5231
Commit 284e5231 authored Mar 28, 2018 by Vadim Pisarevsky
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

tree.cpp modules/ml/src/tree.cpp +1 -0

No files found.
--- 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;
    }
 }