Package org.apache.spark.ml.tree

Class InternalNode

Object
org.apache.spark.ml.tree.Node
org.apache.spark.ml.tree.InternalNode
All Implemented Interfaces:
Serializable, scala.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:

  • Method Details

    • 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