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, Saveable

public class GradientBoostedTreesModel
extends TreeEnsembleModel
implements Saveable

:: 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()
static GradientBoostedTreesModel	load(SparkContext sc, String path)
void	save(SparkContext sc, String path) Save this model to the given path.
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

load

public static GradientBoostedTreesModel load(SparkContext sc,
                             String path)

algo

public scala.Enumeration.Value algo()

trees

public DecisionTreeModel[] trees()

treeWeights

public double[] treeWeights()

save

public void save(SparkContext sc,
        String path)

Description copied from interface: Saveable

Save this model to the given path.

This saves: - human-readable (JSON) model metadata to path/metadata/ - Parquet formatted data to path/data/

The model may be loaded using Loader.load.

Specified by:

save in interface Saveable

Parameters:

sc - Spark context used to save model data.

path - Path specifying the directory in which to save this model. If the directory already exists, this method throws an exception.