hex.tree

Class SharedTree<M extends SharedTreeModel<M,P,O>,P extends SharedTreeModel.SharedTreeParameters,O extends SharedTreeModel.SharedTreeOutput>

java.lang.Object

water.Iced<T>

water.Keyed

water.Job<M>

hex.ModelBuilder<M,P,O>

hex.SupervisedModelBuilder<M,P,O>

hex.tree.SharedTree<M,P,O>

All Implemented Interfaces:

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

Direct Known Subclasses:

DRF, GBM

public abstract class SharedTree<M extends SharedTreeModel<M,P,O>,P extends SharedTreeModel.SharedTreeParameters,O extends SharedTreeModel.SharedTreeOutput>
extends hex.SupervisedModelBuilder<M,P,O>

See Also:

Serialized Form

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
protected class	SharedTree.Driver

Nested classes/interfaces inherited from class hex.ModelBuilder

hex.ModelBuilder.BuilderVisibility, 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

Modifier and Type	Field and Description
protected double	_initialPrediction
protected M	_model
protected int	_ncols
protected int	_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
SharedTree(java.lang.String name, P parms)

Method Summary

Methods

Modifier and Type	Method and Description
protected DHistogram[][][]	buildLayer(water.fvec.Frame fr, int nbins, DTree[] ktrees, int[] leafs, DHistogram[][][] hcs, boolean subset, boolean build_tree_one_node)
protected void	checkMemoryFootPrint()
protected water.fvec.Chunk	chk_nids(water.fvec.Chunk[] chks, int c)
protected water.fvec.Chunk	chk_oobt(water.fvec.Chunk[] chks)
protected water.fvec.Chunk	chk_resp(water.fvec.Chunk[] chks)
protected water.fvec.Chunk	chk_tree(water.fvec.Chunk[] chks, int c)
protected water.fvec.Chunk	chk_work(water.fvec.Chunk[] chks, int c)
static java.util.Random	createRNG(long seed)
protected double[]	data_row(water.fvec.Chunk[] chks, int row, double[] data)
protected double	doScoringAndSaveModel(boolean finalScoring, boolean oob, boolean build_tree_one_node)
protected int	idx_nids(int c)
protected int	idx_oobt()
protected int	idx_resp()
protected int	idx_tree(int c)
protected int	idx_work(int c)
void	init(boolean expensive) Initialize the ModelBuilder, validating all arguments and preparing the training frame.
protected abstract DTree.DecidedNode	makeDecided(DTree.UndecidedNode udn, DHistogram[] hs)
protected static void	printGenerateTrees(DTree[] trees)
protected abstract double	score1(water.fvec.Chunk[] chks, double[] fs, int row)
protected water.fvec.Vec	vec_nids(water.fvec.Frame fr, int t)
protected water.fvec.Vec	vec_resp(water.fvec.Frame fr)
protected water.fvec.Vec	vec_tree(water.fvec.Frame fr, int c)

Methods inherited from class hex.SupervisedModelBuilder

isClassifier, isSupervised, response, vresponse

Methods inherited from class hex.ModelBuilder

builderVisibility, can_build, clearInitState, createModelBuilder, error_count, error, getAlgo, getAlgo, getAlgo, getAlgoFullName, getModelBuilder, getModelBuilders, getModelClass, hide, info, registerModelBuilder, schema, train, trainModel, valid, validationErrors, warn

Methods inherited from class water.Job

cancel, cancel, checksum_impl, dest, done, failed, get, isCancelledOrCrashed, isDone, isRunning, isRunning, isStopped, jobs, msec, onCancelled, progress_msg, progress, remove_impl, start, update, update, update, update

Methods inherited from class water.Keyed

checksum, remove, remove, remove, remove

Methods inherited from class water.Iced

clone, frozenType, read_impl, read, readExternal, readJSON_impl, readJSON, toJsonString, 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

Field Detail

_ntrees

protected int _ntrees

_model

protected M extends SharedTreeModel<M,P,O> _model

_ncols

protected int _ncols

_initialPrediction

protected double _initialPrediction

Constructor Detail

SharedTree

public SharedTree(java.lang.String name,
          P parms)

Method Detail

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 requested ntrees; precompute actual ntrees. Validate the number of classes to predict on; validate a checkpoint.

Overrides:

init in class hex.SupervisedModelBuilder<M extends SharedTreeModel<M,P,O>,P extends SharedTreeModel.SharedTreeParameters,O extends SharedTreeModel.SharedTreeOutput>

buildLayer

protected DHistogram[][][] buildLayer(water.fvec.Frame fr,
                          int nbins,
                          DTree[] ktrees,
                          int[] leafs,
                          DHistogram[][][] hcs,
                          boolean subset,
                          boolean build_tree_one_node)

idx_resp

protected int idx_resp()

idx_oobt

protected int idx_oobt()

idx_tree

protected int idx_tree(int c)

idx_work

protected int idx_work(int c)

idx_nids

protected int idx_nids(int c)

chk_resp

protected water.fvec.Chunk chk_resp(water.fvec.Chunk[] chks)

chk_tree

protected water.fvec.Chunk chk_tree(water.fvec.Chunk[] chks,
                        int c)

chk_work

protected water.fvec.Chunk chk_work(water.fvec.Chunk[] chks,
                        int c)

chk_nids

protected water.fvec.Chunk chk_nids(water.fvec.Chunk[] chks,
                        int c)

chk_oobt

protected water.fvec.Chunk chk_oobt(water.fvec.Chunk[] chks)

vec_nids

protected final water.fvec.Vec vec_nids(water.fvec.Frame fr,
                      int t)

vec_resp

protected final water.fvec.Vec vec_resp(water.fvec.Frame fr)

vec_tree

protected final water.fvec.Vec vec_tree(water.fvec.Frame fr,
                      int c)

data_row

protected double[] data_row(water.fvec.Chunk[] chks,
                int row,
                double[] data)

makeDecided

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

score1

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

doScoringAndSaveModel

protected double doScoringAndSaveModel(boolean finalScoring,
                           boolean oob,
                           boolean build_tree_one_node)

printGenerateTrees

protected static void printGenerateTrees(DTree[] trees)

createRNG

public static java.util.Random createRNG(long seed)

checkMemoryFootPrint

protected void checkMemoryFootPrint()

Overrides:

checkMemoryFootPrint in class hex.ModelBuilder<M extends SharedTreeModel<M,P,O>,P extends SharedTreeModel.SharedTreeParameters,O extends SharedTreeModel.SharedTreeOutput>