org.apache.spark.ml.tree
Class InternalNode

Object
  extended by org.apache.spark.ml.tree.Node
      extended by org.apache.spark.ml.tree.InternalNode
All Implemented Interfaces:
java.io.Serializable

public final class InternalNode
extends Node

:: DeveloperApi :: Internal Decision Tree node. param: prediction Prediction this node would make if it were a leaf node param: impurity Impurity measure at this node (for training data) param: gain Information gain value. Values < 0 indicate missing values; this quirk will be removed with future updates. param: leftChild Left-hand child node param: rightChild Right-hand child node param: split Information about the test used to split to the left or right child.

See Also:
Serialized Form

Method Summary
 double gain()
           
 double impurity()
          Impurity measure at this node (for training data)
 Node leftChild()
           
 double prediction()
          Prediction a leaf node makes, or which an internal node would make if it were a leaf node
 Node rightChild()
           
 Split split()
           
 String toString()
           
 
Methods inherited from class org.apache.spark.ml.tree.Node
fromOld
 
Methods inherited from class Object
equals, getClass, hashCode, notify, notifyAll, wait, wait, wait
 

Method Detail

prediction

public double prediction()
Description copied from class: Node
Prediction a leaf node makes, or which an internal node would make if it were a leaf node

Specified by:
prediction in class Node

impurity

public double impurity()
Description copied from class: Node
Impurity measure at this node (for training data)

Specified by:
impurity in class Node

gain

public double gain()

leftChild

public Node leftChild()

rightChild

public Node rightChild()

split

public Split split()

toString

public String toString()
Overrides:
toString in class Object