From a6718f79aa1c93956e8cae38f32be13f857a99f5 Mon Sep 17 00:00:00 2001 From: Peng Yu Date: Wed, 14 Mar 2018 14:34:28 -0400 Subject: [PATCH 1/2] add impurity stats in debug string --- mllib/src/main/scala/org/apache/spark/ml/tree/Node.scala | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/mllib/src/main/scala/org/apache/spark/ml/tree/Node.scala b/mllib/src/main/scala/org/apache/spark/ml/tree/Node.scala index d30be452a436..76323daf8759 100644 --- a/mllib/src/main/scala/org/apache/spark/ml/tree/Node.scala +++ b/mllib/src/main/scala/org/apache/spark/ml/tree/Node.scala @@ -17,7 +17,7 @@ package org.apache.spark.ml.tree -import org.apache.spark.ml.linalg.Vector +import org.apache.spark.ml.linalg.{Vector, Vectors} import org.apache.spark.mllib.tree.impurity.ImpurityCalculator import org.apache.spark.mllib.tree.model.{ImpurityStats, InformationGainStats => OldInformationGainStats, Node => OldNode, Predict => OldPredict} @@ -122,8 +122,9 @@ class LeafNode private[ml] ( override private[tree] def numDescendants: Int = 0 override private[tree] def subtreeToString(indentFactor: Int = 0): String = { + val stats: Vector = Vectors.dense(impurityStats.stats) val prefix: String = " " * indentFactor - prefix + s"Predict: $prediction\n" + prefix + s"Predict: $prediction, Stats: $stats \n" } override private[tree] def subtreeDepth: Int = 0 From ac0942aadd2a25e87a5c644d4553511e772176fa Mon Sep 17 00:00:00 2001 From: Peng Yu Date: Wed, 14 Mar 2018 14:42:13 -0400 Subject: [PATCH 2/2] fix string --- mllib/src/main/scala/org/apache/spark/ml/tree/Node.scala | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mllib/src/main/scala/org/apache/spark/ml/tree/Node.scala b/mllib/src/main/scala/org/apache/spark/ml/tree/Node.scala index 76323daf8759..af31f16ad10a 100644 --- a/mllib/src/main/scala/org/apache/spark/ml/tree/Node.scala +++ b/mllib/src/main/scala/org/apache/spark/ml/tree/Node.scala @@ -122,7 +122,7 @@ class LeafNode private[ml] ( override private[tree] def numDescendants: Int = 0 override private[tree] def subtreeToString(indentFactor: Int = 0): String = { - val stats: Vector = Vectors.dense(impurityStats.stats) + val stats: String = Vectors.dense(impurityStats.stats).toString val prefix: String = " " * indentFactor prefix + s"Predict: $prediction, Stats: $stats \n" }