org.apache.spark.mllib.tree.model

Class GradientBoostedTreesModel

Object

org.apache.spark.mllib.tree.model.TreeEnsembleModel

org.apache.spark.mllib.tree.model.GradientBoostedTreesModel

All Implemented Interfaces:

java.io.Serializable

public class GradientBoostedTreesModel
extends TreeEnsembleModel

:: Experimental :: Represents a gradient boosted trees model.

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
GradientBoostedTreesModel(scala.Enumeration.Value algo, DecisionTreeModel[] trees, double[] treeWeights)

Method Summary

Methods

Modifier and Type	Method and Description
scala.Enumeration.Value	algo()
DecisionTreeModel[]	trees()
double[]	treeWeights()

Methods inherited from class org.apache.spark.mllib.tree.model.TreeEnsembleModel

numTrees, predict, predict, predict, toDebugString, toString, totalNumNodes

Methods inherited from class Object

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

Constructor Detail

GradientBoostedTreesModel

public GradientBoostedTreesModel(scala.Enumeration.Value algo,
                         DecisionTreeModel[] trees,
                         double[] treeWeights)

Method Detail

algo

public scala.Enumeration.Value algo()

trees

public DecisionTreeModel[] trees()

treeWeights

public double[] treeWeights()