hex.tree.drf

Class DRF

java.lang.Object

water.Iced<T>

water.Keyed

water.Job<M>

hex.ModelBuilder<M,P,O>

hex.tree.SharedTree<DRFModel,DRFModel.DRFParameters,DRFModel.DRFOutput>

hex.tree.drf.DRF

All Implemented Interfaces:

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

public class DRF
extends SharedTree<DRFModel,DRFModel.DRFParameters,DRFModel.DRFOutput>

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.BuilderVisibility

Nested classes/interfaces inherited from class water.Job

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

Nested classes/interfaces inherited from class water.Keyed

water.Keyed.BinarySerializer<X extends water.Keyed>

Field Summary

Fields

Modifier and Type	Field and Description
protected int	_mtry

Fields inherited from class hex.tree.SharedTree

_initialPrediction, _model, _ncols, _ntrees, MAX_NTREES

Fields inherited from class hex.ModelBuilder

_fold, _nclass, _offset, _parms, _response, _train, _valid, _vresponse, _weights

Fields inherited from class water.Job

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

Fields inherited from class water.Keyed

_key, EMPTY_KEY_LIST

Constructor Summary

Constructors

Constructor and Description
DRF(DRFModel.DRFParameters parms)

Method Summary

Methods

Modifier and Type	Method and Description
hex.ModelBuilder.BuilderVisibility	builderVisibility()
hex.ModelCategory[]	can_build()
void	init(boolean expensive) Initialize the ModelBuilder, validating all arguments and preparing the training frame.
protected DTree.DecidedNode	makeDecided(DTree.UndecidedNode udn, DHistogram[] hs)
DRFV3	schema()
protected double	score1(water.fvec.Chunk[] chks, double weight, double offset, double[] fs, int row)
water.Job<DRFModel>	trainModelImpl(long work, boolean restartTimer) Start the DRF training Job on an F/J thread.

Methods inherited from class hex.tree.SharedTree

buildLayer, checkMemoryFootPrint, chk_nids, chk_offset, chk_oobt, chk_resp, chk_tree, chk_weight, chk_work, computePriorClassDistribution, createRNG, data_row, doScoringAndSaveModel, getInitialValue, idx_nids, idx_offset, idx_oobt, idx_resp, idx_tree, idx_weight, idx_work, isSupervised, printGenerateTrees, progressUnits, response, vec_nids, vec_resp, vec_tree, vresponse

Methods inherited from class hex.ModelBuilder

canBeDone, cancel, checkDistributions, clearInitState, computeCrossValidation, createModelBuilder, deleteProgressKey, error_count, getAlgo, getAlgo, getAlgo, getAlgoFullName, getModelBuilder, getModelBuilders, getModelClass, hasFoldCol, hasOffsetCol, hasWeightCol, ignoreBadColumns, ignoreStringColumns, isClassifier, modifyParmsForCrossValidationMainModel, modifyParmsForCrossValidationSplits, nclasses, nFoldCV, numSpecialCols, registerModelBuilder, responseMean, separateFeatureVecs, train, trainModel, updateModelOutput, valid

Methods inherited from class water.Job

block, cancel, checksum_impl, clearValidationErrors, createProgressKey, dest, done, done, error_count_or_uninitialized, error, failed, get, hide, info, isCancelledOrCrashed, isDone, isRunning, isRunning, isStopped, jobKey, jobs, message, msec, progress_msg, progress, remove_impl, start, update, update, update, update, updateValidationMessages, validationErrors, warn

Methods inherited from class water.Keyed

checksum, getBinarySerializer, getPublishedKeys, remove, remove, remove, remove

Methods inherited from class water.Iced

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

Methods inherited from class java.lang.Object

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

Field Detail

_mtry

protected int _mtry

Constructor Detail

DRF

public DRF(DRFModel.DRFParameters parms)

Method Detail

can_build

public hex.ModelCategory[] can_build()

Specified by:

can_build in class hex.ModelBuilder<DRFModel,DRFModel.DRFParameters,DRFModel.DRFOutput>

builderVisibility

public hex.ModelBuilder.BuilderVisibility builderVisibility()

Specified by:

builderVisibility in class hex.ModelBuilder<DRFModel,DRFModel.DRFParameters,DRFModel.DRFOutput>

schema

public DRFV3 schema()

Specified by:

schema in class hex.ModelBuilder<DRFModel,DRFModel.DRFParameters,DRFModel.DRFOutput>

trainModelImpl

public water.Job<DRFModel> trainModelImpl(long work,
                                 boolean restartTimer)

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

Specified by:

trainModelImpl in class hex.ModelBuilder<DRFModel,DRFModel.DRFParameters,DRFModel.DRFOutput>

Parameters:

work -

restartTimer -

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.

Overrides:

init in class SharedTree<DRFModel,DRFModel.DRFParameters,DRFModel.DRFOutput>

makeDecided

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

Specified by:

makeDecided in class SharedTree<DRFModel,DRFModel.DRFParameters,DRFModel.DRFOutput>

score1

protected double score1(water.fvec.Chunk[] chks,
            double weight,
            double offset,
            double[] fs,
            int row)

Specified by:

score1 in class SharedTree<DRFModel,DRFModel.DRFParameters,DRFModel.DRFOutput>