public abstract class SharedTree<M extends SharedTreeModel<M,P,O>,P extends SharedTreeModel.SharedTreeParameters,O extends SharedTreeModel.SharedTreeOutput>
extends hex.ModelBuilder<M,P,O>
Modifier and Type | Class and Description |
---|---|
protected class |
SharedTree.Driver |
hex.ModelBuilder.BuilderVisibility, hex.ModelBuilder.ValidationMessage
Modifier and Type | Field and Description |
---|---|
protected double |
_initialPrediction |
protected M |
_model |
protected int |
_ncols |
protected int |
_ntrees |
_messages, _nclass, _offset, _parms, _response, _train, _valid, _vresponse, _weights
Constructor and Description |
---|
SharedTree(java.lang.String name,
P parms) |
Modifier and Type | Method and Description |
---|---|
protected DHistogram[][][] |
buildLayer(water.fvec.Frame fr,
int nbins,
int nbins_cats,
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) |
protected boolean |
computePriorClassDistribution() |
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.
|
boolean |
isSupervised() |
protected abstract DTree.DecidedNode |
makeDecided(DTree.UndecidedNode udn,
DHistogram[] hs) |
protected static void |
printGenerateTrees(DTree[] trees) |
water.fvec.Vec |
response() |
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) |
water.fvec.Vec |
vresponse() |
builderVisibility, can_build, clearInitState, createModelBuilder, error_count, error, getAlgo, getAlgo, getAlgo, getAlgoFullName, getModelBuilder, getModelBuilders, getModelClass, hasOffset, hasWeights, hide, ignoreBadColumns, ignoreStringColumns, info, isClassifier, nclasses, registerModelBuilder, schema, separateFeatureVecs, train, trainModel, updateValidationMessages, valid, validationErrors, warn
cancel, cancel, checksum_impl, createProgressKey, deleteProgressKey, dest, done, failed, get, isCancelledOrCrashed, isDone, isRunning, isRunning, isStopped, jobs, msec, onCancelled, progress_msg, progress, remove_impl, start, update, update, update, update
checksum, getBinarySerializer, getPublishedKeys, remove, remove, remove, remove
protected int _ntrees
protected M extends SharedTreeModel<M,P,O> _model
protected int _ncols
protected double _initialPrediction
public SharedTree(java.lang.String name, P parms)
public boolean isSupervised()
isSupervised
in class hex.ModelBuilder<M extends SharedTreeModel<M,P,O>,P extends SharedTreeModel.SharedTreeParameters,O extends SharedTreeModel.SharedTreeOutput>
public water.fvec.Vec response()
response
in class hex.ModelBuilder<M extends SharedTreeModel<M,P,O>,P extends SharedTreeModel.SharedTreeParameters,O extends SharedTreeModel.SharedTreeOutput>
public water.fvec.Vec vresponse()
vresponse
in class hex.ModelBuilder<M extends SharedTreeModel<M,P,O>,P extends SharedTreeModel.SharedTreeParameters,O extends SharedTreeModel.SharedTreeOutput>
protected boolean computePriorClassDistribution()
computePriorClassDistribution
in class hex.ModelBuilder<M extends SharedTreeModel<M,P,O>,P extends SharedTreeModel.SharedTreeParameters,O extends SharedTreeModel.SharedTreeOutput>
public void init(boolean expensive)
init
in class hex.ModelBuilder<M extends SharedTreeModel<M,P,O>,P extends SharedTreeModel.SharedTreeParameters,O extends SharedTreeModel.SharedTreeOutput>
protected DHistogram[][][] buildLayer(water.fvec.Frame fr, int nbins, int nbins_cats, DTree[] ktrees, int[] leafs, DHistogram[][][] hcs, boolean subset, boolean build_tree_one_node)
protected int idx_resp()
protected int idx_oobt()
protected int idx_tree(int c)
protected int idx_work(int c)
protected int idx_nids(int c)
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)
protected water.fvec.Chunk chk_nids(water.fvec.Chunk[] chks, int c)
protected water.fvec.Chunk chk_oobt(water.fvec.Chunk[] chks)
protected final water.fvec.Vec vec_nids(water.fvec.Frame fr, int t)
protected final water.fvec.Vec vec_resp(water.fvec.Frame fr)
protected final water.fvec.Vec vec_tree(water.fvec.Frame fr, int c)
protected double[] data_row(water.fvec.Chunk[] chks, int row, double[] data)
protected abstract DTree.DecidedNode makeDecided(DTree.UndecidedNode udn, DHistogram[] hs)
protected abstract double score1(water.fvec.Chunk[] chks, double[] fs, int row)
protected double doScoringAndSaveModel(boolean finalScoring, boolean oob, boolean build_tree_one_node)
protected static void printGenerateTrees(DTree[] trees)
public static java.util.Random createRNG(long seed)
protected void checkMemoryFootPrint()
checkMemoryFootPrint
in class hex.ModelBuilder<M extends SharedTreeModel<M,P,O>,P extends SharedTreeModel.SharedTreeParameters,O extends SharedTreeModel.SharedTreeOutput>