org.apache.spark.ml.tree

Class InternalNode

Object

org.apache.spark.ml.tree.Node

org.apache.spark.ml.tree.InternalNode

All Implemented Interfaces:

java.io.Serializable

public class InternalNode
extends Node

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 less than 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

Modifier and Type	Method and Description
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