org.apache.spark.mllib.tree

Class DecisionTree

Object

org.apache.spark.mllib.tree.DecisionTree

All Implemented Interfaces:

java.io.Serializable, Logging

public class DecisionTree
extends Object
implements scala.Serializable, Logging

:: Experimental :: A class that implements a decision tree algorithm for classification and regression. It supports both continuous and categorical features.

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
DecisionTree(Strategy strategy)

Method Summary

Methods

Modifier and Type	Method and Description
DecisionTreeModel	train(RDD<LabeledPoint> input) Method to train a decision tree model over an RDD

Methods inherited from class Object

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

Methods inherited from interface org.apache.spark.Logging

initialized, initializeIfNecessary, initializeLogging, initLock, isTraceEnabled, log_, log, logDebug, logDebug, logError, logError, logInfo, logInfo, logTrace, logTrace, logWarning, logWarning

Constructor Detail

DecisionTree

public DecisionTree(Strategy strategy)

Method Detail

train

public DecisionTreeModel train(RDD<LabeledPoint> input)

Method to train a decision tree model over an RDD

Parameters:

input - RDD of LabeledPoint used as training data

Returns:

a DecisionTreeModel that can be used for prediction