fix bug

apache · Sep 9, 2014 · 987cbf4 · 987cbf4
1 parent ff34845
commit 987cbf4
Showing 1 changed file with 1 addition and 1 deletion.
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
@@ -145,7 +145,7 @@ class DecisionTree (private val strategy: Strategy) extends Serializable with Lo
         val stats = nodeSplitStats._2
         val predict = nodeSplitStats._3
         val isLeaf = (stats.gain <= 0) || (level == strategy.maxDepth)
-        val node = new Node(nodeIndex, predict, isLeaf, Some(split), None, None, Some(stats))
+        val node = new Node(nodeIndex, predict.predict, isLeaf, Some(split), None, None, Some(stats))
         logDebug("Node = " + node)
         nodes(nodeIndex) = node
         timer.stop("extractNodeInfo")