Package org.apache.spark.ml.tree

Class DecisionTreeModelReadWrite.SplitData

Object
org.apache.spark.ml.tree.DecisionTreeModelReadWrite.SplitData
All Implemented Interfaces:
Serializable, scala.Equals, scala.Product
Enclosing class:
DecisionTreeModelReadWrite
public static class DecisionTreeModelReadWrite.SplitData extends Object implements scala.Product, Serializable
Info for a Split

param: featureIndex Index of feature split on param: leftCategoriesOrThreshold For categorical feature, set of leftCategories. For continuous feature, threshold. param: numCategories For categorical feature, number of categories. For continuous feature, -1.

See Also:

  • Constructor Summary

    Constructors
    Constructor
    Description
    SplitData(int featureIndex, double[] leftCategoriesOrThreshold, int numCategories)
     

  • Method Summary

    Modifier and Type
    Method
    Description
    int
    featureIndex()
     
    Split
    getSplit()
     
    double[]
    leftCategoriesOrThreshold()
     
    int
    numCategories()
     

    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, productElementName, productElementNames, productIterator, productPrefix

  • Constructor Details

    • SplitData

      public SplitData(int featureIndex, double[] leftCategoriesOrThreshold, int numCategories)

  • Method Details

    • featureIndex

      public int featureIndex()

    • leftCategoriesOrThreshold

      public double[] leftCategoriesOrThreshold()

    • numCategories

      public int numCategories()

    • getSplit

      public Split getSplit()