hex.tree

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

java.lang.Object

water.Iced<T>

water.Keyed<T>

water.Lockable<M>

hex.Model<M,P,O>

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

All Implemented Interfaces:

hex.Model.GetMostImportantFeatures, hex.Model.LeafNodeAssignment, java.io.Externalizable, java.io.Serializable, java.lang.Cloneable, water.Freezable<M>

Direct Known Subclasses:

DRFModel, GBMModel, IsolationForestModel

public abstract class SharedTreeModel<M extends SharedTreeModel<M,P,O>,P extends SharedTreeModel.SharedTreeParameters,O extends SharedTreeModel.SharedTreeOutput>
extends hex.Model<M,P,O>
implements hex.Model.LeafNodeAssignment, hex.Model.GetMostImportantFeatures

See Also:

Serialized Form

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	SharedTreeModel.BufStringDecisionPathTracker
static class	SharedTreeModel.SharedTreeOutput
static class	SharedTreeModel.SharedTreeParameters

Nested classes/interfaces inherited from class hex.Model

hex.Model.BigScore, hex.Model.BigScoreChunkPredict, hex.Model.BigScorePredict, hex.Model.DeepFeatures, hex.Model.ExemplarMembers, hex.Model.GetMostImportantFeatures, hex.Model.GLRMArchetypes, hex.Model.GridSortBy, hex.Model.InteractionBuilder, hex.Model.InteractionPair, hex.Model.InteractionSpec, hex.Model.JavaModelStreamWriter, hex.Model.LeafNodeAssignment, hex.Model.Output, hex.Model.Parameters, hex.Model.StagedPredictions

Nested classes/interfaces inherited from interface hex.Model.LeafNodeAssignment

hex.Model.LeafNodeAssignment.LeafNodeAssignmentType

Field Summary

Fields inherited from class hex.Model

_dist, _output, _parms, _toDelete, _warnings, _warningsP, scoringInfo

Fields inherited from class water.Lockable

_lockers

Fields inherited from class water.Keyed

_key

Constructor Summary

Constructors

Constructor and Description
SharedTreeModel(water.Key<M> selfKey, P parms, O output)

Method Summary

Modifier and Type	Method and Description
protected boolean	binomialOpt()
protected M	deepClone(water.Key<M> result) Performs deep clone of given model.
java.lang.String[]	getMostImportantFeatures(int n)
hex.genmodel.algos.tree.SharedTreeSubgraph	getSharedTreeSubgraph(int tidx, int cls) Converts a given tree of the ensemble to a user-understandable representation.
hex.ToEigenVec	getToEigenVec()
protected java.lang.String[]	makeAllTreeColumnNames()
hex.ModelMetrics.MetricBuilder	makeMetricBuilder(java.lang.String[] domain)
protected water.fvec.Frame	postProcessPredictions(water.fvec.Frame adaptedFrame, water.fvec.Frame predictFr, water.Job j)
protected water.Keyed	readAll_impl(water.AutoBuffer ab, water.Futures fs)
protected water.Futures	remove_impl(water.Futures fs)
protected double[]	score0(double[] data, double[] preds)
protected double[]	score0(double[] data, double[] preds, double offset)
protected double[]	score0(double[] data, double[] preds, double offset, int ntrees)
protected double[]	score0(double[] data, double[] preds, double offset, int startTree, int ntrees)
protected double[]	score0Incremental(Score.ScoreIncInfo sii, water.fvec.Chunk[] chks, double offset, int row_in_chunk, double[] tmp, double[] preds)
water.fvec.Frame	scoreLeafNodeAssignment(water.fvec.Frame frame, hex.Model.LeafNodeAssignment.LeafNodeAssignmentType type, water.Key<water.fvec.Frame> destination_key)
protected boolean	toJavaCheckTooBig()
protected <T extends water.exceptions.JCodeSB> T	toJavaForestName(T sb, java.lang.String mname, int t)
protected water.util.SBPrintStream	toJavaInit(water.util.SBPrintStream sb, water.codegen.CodeGeneratorPipeline fileCtx)
protected void	toJavaPredictBody(water.util.SBPrintStream body, water.codegen.CodeGeneratorPipeline classCtx, water.codegen.CodeGeneratorPipeline fileCtx, boolean verboseCode)
protected <T extends water.exceptions.JCodeSB> T	toJavaTreeName(T sb, java.lang.String mname, int t, int c)
protected abstract void	toJavaUnifyPreds(water.util.SBPrintStream body)
protected water.AutoBuffer	writeAll_impl(water.AutoBuffer ab) Write out K/V pairs

Methods inherited from class hex.Model

adaptTestForTrain, adaptTestForTrain, addMetrics, addModelMetrics, addWarning, addWarningP, auc, checksum_impl, classification_error, compareTo, computeDeviances, containsResponse, data, defaultThreshold, defaultThreshold, deleteCrossValidationModels, deleteCrossValidationPreds, deviance, deviance, exportBinaryModel, exportMojo, fetchAll, fillScoringInfo, getDefaultGridSortBy, getMojo, getPojoInterfaces, haveMojo, havePojo, importBinaryModel, isSupervised, last_scored, lift_top_group, logloss, loss, mae, makeBigScoreTask, makeInteraction, makeInteractions, makeInteractions, makeSchema, makeScoringDomains, makeScoringNames, makeScoringNames, mean_per_class_error, mse, needsPostProcess, predictScoreImpl, r2, rmsle, score, score, score, score, score, score, score, score0, score0, score0PostProcessSupervised, scoreMetrics, scoring_history, scoringDomains, setupBigScorePredict, testJavaScoring, testJavaScoring, testJavaScoring, testJavaScoring, toJava, toJava, toJava, toJavaNCLASSES, toJavaPROB, toJavaSuper, toJavaTransform, toMojo, toString

Methods inherited from class water.Lockable

delete_and_lock, delete_and_lock, delete_and_lock, delete, delete, delete, read_lock, read_lock, read_lock, unlock_all, unlock, unlock, unlock, unlock, update, update, update, write_lock, write_lock, write_lock

Methods inherited from class water.Keyed

checksum, readAll, remove, remove, remove, remove, writeAll

Methods inherited from class water.Iced

asBytes, clone, copyOver, frozenType, read, readExternal, readJSON, reloadFromBytes, toJsonString, write, writeExternal, writeJSON

Methods inherited from class java.lang.Object

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

Constructor Detail

SharedTreeModel

public SharedTreeModel(water.Key<M> selfKey,
                       P parms,
                       O output)

Method Detail

getMostImportantFeatures

public java.lang.String[] getMostImportantFeatures(int n)

Specified by:

getMostImportantFeatures in interface hex.Model.GetMostImportantFeatures

getToEigenVec

public hex.ToEigenVec getToEigenVec()

Overrides:

getToEigenVec in class hex.Model<M extends SharedTreeModel<M,P,O>,P extends SharedTreeModel.SharedTreeParameters,O extends SharedTreeModel.SharedTreeOutput>

makeMetricBuilder

public hex.ModelMetrics.MetricBuilder makeMetricBuilder(java.lang.String[] domain)

Specified by:

makeMetricBuilder in class hex.Model<M extends SharedTreeModel<M,P,O>,P extends SharedTreeModel.SharedTreeParameters,O extends SharedTreeModel.SharedTreeOutput>

makeAllTreeColumnNames

protected java.lang.String[] makeAllTreeColumnNames()

scoreLeafNodeAssignment

public water.fvec.Frame scoreLeafNodeAssignment(water.fvec.Frame frame,
                                                hex.Model.LeafNodeAssignment.LeafNodeAssignmentType type,
                                                water.Key<water.fvec.Frame> destination_key)

Specified by:

scoreLeafNodeAssignment in interface hex.Model.LeafNodeAssignment

postProcessPredictions

protected water.fvec.Frame postProcessPredictions(water.fvec.Frame adaptedFrame,
                                                  water.fvec.Frame predictFr,
                                                  water.Job j)

Overrides:

postProcessPredictions in class hex.Model<M extends SharedTreeModel<M,P,O>,P extends SharedTreeModel.SharedTreeParameters,O extends SharedTreeModel.SharedTreeOutput>

score0Incremental

protected double[] score0Incremental(Score.ScoreIncInfo sii,
                                     water.fvec.Chunk[] chks,
                                     double offset,
                                     int row_in_chunk,
                                     double[] tmp,
                                     double[] preds)

score0

protected double[] score0(double[] data,
                          double[] preds,
                          double offset)

Overrides:

score0 in class hex.Model<M extends SharedTreeModel<M,P,O>,P extends SharedTreeModel.SharedTreeParameters,O extends SharedTreeModel.SharedTreeOutput>

score0

protected double[] score0(double[] data,
                          double[] preds)

Specified by:

score0 in class hex.Model<M extends SharedTreeModel<M,P,O>,P extends SharedTreeModel.SharedTreeParameters,O extends SharedTreeModel.SharedTreeOutput>

score0

protected double[] score0(double[] data,
                          double[] preds,
                          double offset,
                          int ntrees)

score0

protected double[] score0(double[] data,
                          double[] preds,
                          double offset,
                          int startTree,
                          int ntrees)

deepClone

protected M deepClone(water.Key<M> result)

Performs deep clone of given model.

remove_impl

protected water.Futures remove_impl(water.Futures fs)

Overrides:

remove_impl in class hex.Model<M extends SharedTreeModel<M,P,O>,P extends SharedTreeModel.SharedTreeParameters,O extends SharedTreeModel.SharedTreeOutput>

writeAll_impl

protected water.AutoBuffer writeAll_impl(water.AutoBuffer ab)

Write out K/V pairs

Overrides:

writeAll_impl in class hex.Model<M extends SharedTreeModel<M,P,O>,P extends SharedTreeModel.SharedTreeParameters,O extends SharedTreeModel.SharedTreeOutput>

readAll_impl

protected water.Keyed readAll_impl(water.AutoBuffer ab,
                                   water.Futures fs)

Overrides:

readAll_impl in class hex.Model<M extends SharedTreeModel<M,P,O>,P extends SharedTreeModel.SharedTreeParameters,O extends SharedTreeModel.SharedTreeOutput>

getSharedTreeSubgraph

public hex.genmodel.algos.tree.SharedTreeSubgraph getSharedTreeSubgraph(int tidx,
                                                                        int cls)

Converts a given tree of the ensemble to a user-understandable representation.

Parameters:

tidx - tree index

cls - tree class

Returns:

instance of SharedTreeSubgraph

toJavaCheckTooBig

protected boolean toJavaCheckTooBig()

Overrides:

toJavaCheckTooBig in class hex.Model<M extends SharedTreeModel<M,P,O>,P extends SharedTreeModel.SharedTreeParameters,O extends SharedTreeModel.SharedTreeOutput>

binomialOpt

protected boolean binomialOpt()

toJavaInit

protected water.util.SBPrintStream toJavaInit(water.util.SBPrintStream sb,
                                              water.codegen.CodeGeneratorPipeline fileCtx)

Overrides:

toJavaInit in class hex.Model<M extends SharedTreeModel<M,P,O>,P extends SharedTreeModel.SharedTreeParameters,O extends SharedTreeModel.SharedTreeOutput>

toJavaPredictBody

protected void toJavaPredictBody(water.util.SBPrintStream body,
                                 water.codegen.CodeGeneratorPipeline classCtx,
                                 water.codegen.CodeGeneratorPipeline fileCtx,
                                 boolean verboseCode)

Overrides:

toJavaPredictBody in class hex.Model<M extends SharedTreeModel<M,P,O>,P extends SharedTreeModel.SharedTreeParameters,O extends SharedTreeModel.SharedTreeOutput>

toJavaUnifyPreds

protected abstract void toJavaUnifyPreds(water.util.SBPrintStream body)

toJavaTreeName

protected <T extends water.exceptions.JCodeSB> T toJavaTreeName(T sb,
                                                                java.lang.String mname,
                                                                int t,
                                                                int c)

toJavaForestName

protected <T extends water.exceptions.JCodeSB> T toJavaForestName(T sb,
                                                                  java.lang.String mname,
                                                                  int t)