org.apache.spark.ml.tree

Class DecisionTreeModelReadWrite.NodeData

Object

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

All Implemented Interfaces:

java.io.Serializable, scala.Equals, scala.Product

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: 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 and Description
NodeData(int id, double prediction, double impurity, double[] impurityStats, double gain, int leftChild, int rightChild, DecisionTreeModelReadWrite.SplitData split)

Method Summary

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

Methods inherited from class Object

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

Methods inherited from interface scala.Product

productArity, productElement, productIterator, productPrefix

Methods inherited from interface scala.Equals

canEqual, equals

Constructor Detail

NodeData

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

Method Detail

id

public int id()

prediction

public double prediction()

impurity

public double impurity()

impurityStats

public double[] impurityStats()

gain

public double gain()

leftChild

public int leftChild()

rightChild

public int rightChild()

split

public DecisionTreeModelReadWrite.SplitData split()