fixing while loop

manishamde · manishamde · commit f16a9bb09d65 · 2014-05-17T23:45:04.000-07:00
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
               }

Original file line number	Diff line number	Diff line change
`@@ -941,6 +941,7 @@ object DecisionTree extends Serializable with Logging {`
`941`	`941`	`while (classIndex < numClasses) {`
`942`	`942`	`rightNodeAgg(featureIndex)(splitIndex)(classIndex)`
`943`	`943`	`= totalNodeAgg(classIndex) - leftNodeAgg(featureIndex)(splitIndex)(classIndex)`
	`944`	`+ classIndex += 1`
`944`	`945`	`}`
`945`	`946`	`splitIndex += 1`
`946`	`947`	`}`