org.apache.spark.mllib.tree.model

Class DecisionTreeModel

Object

org.apache.spark.mllib.tree.model.DecisionTreeModel

All Implemented Interfaces:

java.io.Serializable

public class DecisionTreeModel
extends Object
implements scala.Serializable

:: Experimental :: Decision tree model for classification or regression. This model stores the decision tree structure and parameters.

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
DecisionTreeModel(Node topNode, scala.Enumeration.Value algo)

Method Summary

Methods

Modifier and Type	Method and Description
scala.Enumeration.Value	algo()
int	depth() Get depth of tree.
int	numNodes() Get number of nodes in tree, including leaf nodes.
RDD<Object>	predict(RDD<Vector> features) Predict values for the given data set using the model trained.
double	predict(Vector features) Predict values for a single data point using the model trained.
Node	topNode()
String	toString() Print full model.

Methods inherited from class Object

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

Constructor Detail

DecisionTreeModel

public DecisionTreeModel(Node topNode,
                 scala.Enumeration.Value algo)

Method Detail

topNode

public Node topNode()

algo

public scala.Enumeration.Value algo()

predict

public double predict(Vector features)

Predict values for a single data point using the model trained.

Parameters:

features - array representing a single data point

Returns:

Double prediction from the trained model

predict

public RDD<Object> predict(RDD<Vector> features)

Predict values for the given data set using the model trained.

Parameters:

features - RDD representing data points to be predicted

Returns:

RDD[Int] where each entry contains the corresponding prediction

numNodes

public int numNodes()

Get number of nodes in tree, including leaf nodes.

depth

public int depth()

Get depth of tree. E.g.: Depth 0 means 1 leaf node. Depth 1 means 1 internal node and 2 leaf nodes.

toString

public String toString()

Print full model.

Overrides:

toString in class Object