fixing while loop

markhamstra · May 18, 2014 · f16a9bb · f16a9bb
1 parent d811425
commit f16a9bb
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/mllib/src/main/scala/org/apache/spark/mllib/tree/DecisionTree.scala b/mllib/src/main/scala/org/apache/spark/mllib/tree/DecisionTree.scala
@@ -941,6 +941,7 @@ object DecisionTree extends Serializable with Logging {
                 while (classIndex < numClasses) {
                   rightNodeAgg(featureIndex)(splitIndex)(classIndex)
                     = totalNodeAgg(classIndex) - leftNodeAgg(featureIndex)(splitIndex)(classIndex)
+                  classIndex += 1
                 }
                 splitIndex += 1
               }