fix minor bug in RF (#2089)

oneapi-src · Mar 4, 2022 · fc46f5e · fc46f5e
1 parent 4051164
commit fc46f5e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.../src/algorithms/dtrees/forest/classification/df_classification_train_dense_default_impl.i b/.../src/algorithms/dtrees/forest/classification/df_classification_train_dense_default_impl.i
@@ -386,7 +386,7 @@ bool UnorderedRespHelper<algorithmFPType, cpu>::findBestSplitOrderedFeature(cons
     {
         const algorithmFPType weights = this->_aWeights[aIdx[i]].val;
         const bool bSameFeaturePrev(featureVal[i] <= featureVal[i - 1] + accuracy);
-        leftWeights += weights;
+        leftWeights += this->_aWeights[aIdx[i - 1]].val;
         if (bSameFeaturePrev || (i < nMinSplitPart) || (leftWeights < minWeightLeaf) || (totalWeights - leftWeights < minWeightLeaf))
         {
             //can't make a split