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 :: Model to store the decision tree 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()
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()

Methods inherited from class Object

equals, getClass, hashCode, notify, notifyAll, toString, 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