org.apache.spark.mllib.tree.impl

Class TreePoint

Object

org.apache.spark.mllib.tree.impl.TreePoint

All Implemented Interfaces:

java.io.Serializable

public class TreePoint
extends Object
implements scala.Serializable

Internal representation of LabeledPoint for DecisionTree. This bins feature values based on a subsampled of data as follows: (a) Continuous features are binned into ranges. (b) Unordered categorical features are binned based on subsets of feature values. "Unordered categorical features" are categorical features with low arity used in multiclass classification. (c) Ordered categorical features are binned based on feature values. "Ordered categorical features" are categorical features with high arity, or any categorical feature used in regression or binary classification.

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
TreePoint(double label, int[] binnedFeatures)

Method Summary

Methods

Modifier and Type	Method and Description
int[]	binnedFeatures()
static RDD<TreePoint>	convertToTreeRDD(RDD<LabeledPoint> input, Bin[][] bins, DecisionTreeMetadata metadata) Convert an input dataset into its TreePoint representation, binning feature values in preparation for DecisionTree training.
double	label()

Methods inherited from class Object

equals, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

TreePoint

public TreePoint(double label,
         int[] binnedFeatures)

Method Detail

convertToTreeRDD

public static RDD<TreePoint> convertToTreeRDD(RDD<LabeledPoint> input,
                              Bin[][] bins,
                              DecisionTreeMetadata metadata)

Convert an input dataset into its TreePoint representation, binning feature values in preparation for DecisionTree training.

Parameters:

input - Input dataset.

bins - Bins for features, of size (numFeatures, numBins).

metadata - Learning and dataset metadata

Returns:

TreePoint dataset representation

label

public double label()

binnedFeatures

public int[] binnedFeatures()