Package org.apache.spark.ml.tree

Class DecisionTreeModelReadWrite.NodeData

Object

org.apache.spark.ml.tree.DecisionTreeModelReadWrite.NodeData

All Implemented Interfaces:

Serializable, scala.Equals, scala.Product, scala.Serializable

Enclosing class:

DecisionTreeModelReadWrite

public static class DecisionTreeModelReadWrite.NodeData
extends Object
implements scala.Product, scala.Serializable

Info for a Node

param: id Index used for tree reconstruction. Indices follow a pre-order traversal. param: impurityStats Stats array. Impurity type is stored in metadata. param: rawCount The unweighted number of samples falling in this node. param: gain Gain, or arbitrary value if leaf node. param: leftChild Left child index, or arbitrary value if leaf node. param: rightChild Right child index, or arbitrary value if leaf node. param: split Split info, or arbitrary value if leaf node.

See Also:

• Serialized Form

Constructor Summary

Constructors

Constructor	Description
NodeData(int id, double prediction, double impurity, double[] impurityStats, long rawCount, double gain, int leftChild, int rightChild, DecisionTreeModelReadWrite.SplitData split)

Method Summary

Modifier and Type	Method	Description
double	gain()
int	id()
double	impurity()
double[]	impurityStats()
int	leftChild()
double	prediction()
long	rawCount()
int	rightChild()
DecisionTreeModelReadWrite.SplitData	split()

Methods inherited from class java.lang.Object

equals, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Methods inherited from interface scala.Equals

canEqual, equals

Methods inherited from interface scala.Product

productArity, productElement, productIterator, productPrefix

Constructor Details

NodeData

public NodeData(int id,
 double prediction,
 double impurity,
 double[] impurityStats,
 long rawCount,
 double gain,
 int leftChild,
 int rightChild,
 DecisionTreeModelReadWrite.SplitData split)

Method Details

id

public int id()

prediction

public double prediction()

impurity

public double impurity()

impurityStats

public double[] impurityStats()

rawCount

public long rawCount()

gain

public double gain()

leftChild

public int leftChild()

rightChild

public int rightChild()

split

public DecisionTreeModelReadWrite.SplitData split()