From f16a9bb09d6585fa0963b3753f58cd679a4a2f32 Mon Sep 17 00:00:00 2001
From: Manish Amde <manish9ue@gmail.com>
Date: Sat, 17 May 2014 23:45:04 -0700
Subject: [PATCH] fixing while loop

---
 .../main/scala/org/apache/spark/mllib/tree/DecisionTree.scala    | 1 +
 1 file changed, 1 insertion(+)

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
index 6539a1194116c..9ebb1d25ffa02 100644
--- 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
               }