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, scala.Serializable

Enclosing class:

DecisionTreeModelReadWrite

public static class DecisionTreeModelReadWrite.SplitData
extends Object
implements scala.Product, scala.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:

• Serialized Form

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, 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()