hex.tree.gbm

Class GBM

java.lang.Object

water.Iced

water.Keyed

water.Job<M>

hex.ModelBuilder<M,P,O>

hex.SupervisedModelBuilder<M,P,O>

hex.tree.SharedTree<GBMModel,GBMModel.GBMParameters,GBMModel.GBMOutput>

hex.tree.gbm.GBM

All Implemented Interfaces:

java.io.Externalizable, java.io.Serializable, java.lang.Cloneable, water.Freezable

public class GBM
extends SharedTree<GBMModel,GBMModel.GBMParameters,GBMModel.GBMOutput>

Gradient Boosted Trees Based on "Elements of Statistical Learning, Second Edition, page 387"

See Also:

Serialized Form

Nested Class Summary

Nested classes/interfaces inherited from class hex.tree.SharedTree

SharedTree.Driver

Nested classes/interfaces inherited from class hex.ModelBuilder

hex.ModelBuilder.ValidationMessage

Nested classes/interfaces inherited from class water.Job

water.Job.JobCancelledException, water.Job.JobState, water.Job.Progress, water.Job.ProgressUpdate

Field Summary

Fields inherited from class hex.tree.SharedTree

_initialPrediction, _model, _ncols, _ntrees

Fields inherited from class hex.SupervisedModelBuilder

_nclass, _response, _response_key, _vresponse, _vresponse_key

Fields inherited from class hex.ModelBuilder

_messages, _parms, _train, _valid

Fields inherited from class water.Job

_description, _dest, _end_time, _exception, _progressKey, _start_time, _state

Fields inherited from class water.Keyed

_key

Constructor Summary

Constructors

Constructor and Description
GBM(GBMModel.GBMParameters parms)

Method Summary

Methods

Modifier and Type	Method and Description
protected hex.VarImp	doVarImpCalc(boolean scale)
void	init(boolean expensive) Initialize the ModelBuilder, validating all arguments and preparing the training frame.
protected DTree.DecidedNode	makeDecided(DTree.UndecidedNode udn, DHistogram[] hs)
protected boolean	outOfBagRow(water.fvec.Chunk[] chks, int row) Which rows are in-bag vs out-of-bag for sampling
GBMV2	schema()
protected float	score1(water.fvec.Chunk[] chks, float[] fs, int row)
water.Job<GBMModel>	trainModel() Start the GBM training Job on an F/J thread.

Methods inherited from class hex.tree.SharedTree

buildLayer, chk_nids, chk_oobt, chk_resp, chk_tree, chk_work, data_row, doScoringAndSaveModel, printGenerateTrees, vec_nids, vec_resp, vec_tree

Methods inherited from class hex.SupervisedModelBuilder

isClassifier, response, vresponse

Methods inherited from class hex.ModelBuilder

clearInitState, createModelBuilder, error_count, error, getAlgo, getModelBuilder, getModelBuilderName, getModelBuilders, getModelClass, hide, info, registerModelBuilder, train, valid, validationErrors, warn

Methods inherited from class water.Job

cancel, cancel, cancel2, checksum, dest, done, get, isCancelledOrCrashed, isDone, isRunning, isRunning, isStopped, jobs, msec, onCancelled, progress, remove_impl, start, update, update

Methods inherited from class water.Keyed

remove, remove, remove, remove

Methods inherited from class water.Iced

clone, frozenType, read_impl, read, readExternal, readJSON_impl, readJSON, write_impl, write, writeExternal, writeHTML_impl, writeHTML, writeJSON_impl, writeJSON

Methods inherited from class java.lang.Object

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

Constructor Detail

GBM

public GBM(GBMModel.GBMParameters parms)

Method Detail

schema

public GBMV2 schema()

Specified by:

schema in class hex.ModelBuilder<GBMModel,GBMModel.GBMParameters,GBMModel.GBMOutput>

trainModel

public water.Job<GBMModel> trainModel()

Start the GBM training Job on an F/J thread.

Specified by:

trainModel in class hex.ModelBuilder<GBMModel,GBMModel.GBMParameters,GBMModel.GBMOutput>

init

public void init(boolean expensive)

Initialize the ModelBuilder, validating all arguments and preparing the training frame. This call is expected to be overridden in the subclasses and each subclass will start with "super.init();". This call is made by the front-end whenever the GUI is clicked, and needs to be fast; heavy-weight prep needs to wait for the trainModel() call. Validate the learning rate and loss family.

Overrides:

init in class SharedTree<GBMModel,GBMModel.GBMParameters,GBMModel.GBMOutput>

makeDecided

protected DTree.DecidedNode makeDecided(DTree.UndecidedNode udn,
                            DHistogram[] hs)

Specified by:

makeDecided in class SharedTree<GBMModel,GBMModel.GBMParameters,GBMModel.GBMOutput>

outOfBagRow

protected boolean outOfBagRow(water.fvec.Chunk[] chks,
                  int row)

Description copied from class: SharedTree

Which rows are in-bag vs out-of-bag for sampling

Specified by:

outOfBagRow in class SharedTree<GBMModel,GBMModel.GBMParameters,GBMModel.GBMOutput>

doVarImpCalc

protected hex.VarImp doVarImpCalc(boolean scale)

Specified by:

doVarImpCalc in class SharedTree<GBMModel,GBMModel.GBMParameters,GBMModel.GBMOutput>

score1

protected float score1(water.fvec.Chunk[] chks,
           float[] fs,
           int row)

Specified by:

score1 in class SharedTree<GBMModel,GBMModel.GBMParameters,GBMModel.GBMOutput>