Skip to content

add impurity stats in tree leaf node debug string #20825

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
yupbank wants to merge 2 commits into from
Closed

add impurity stats in tree leaf node debug string #20825

Changes from 1 commit
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
a6718f7
add impurity stats in debug string
yupbank Mar 14, 2018
ac0942a
fix string
yupbank Mar 14, 2018
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Next Next commit
add impurity stats in debug string
  • Loading branch information
@yupbank
yupbank committed Mar 14, 2018
commit a6718f79aa1c93956e8cae38f32be13f857a99f5
5 changes: 3 additions & 2 deletions mllib/src/main/scala/org/apache/spark/ml/tree/Node.scala
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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}

Expand Down Expand Up @@ -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
Expand Down