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

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.SharedTreeOutput
static class	SharedTreeModel.SharedTreeParameters

Nested classes/interfaces inherited from class hex.Model

hex.Model.BigScore, hex.Model.DeepFeatures, hex.Model.ExemplarMembers, hex.Model.GetMostImportantFeatures, hex.Model.GLRMArchetypes, hex.Model.GridSortBy, hex.Model.InteractionPair, hex.Model.JavaModelStreamWriter, hex.Model.LeafNodeAssignment, hex.Model.Output, hex.Model.Parameters

Field Summary

Fields inherited from class hex.Model

_dist, _output, _parms, _toDelete, _warnings, 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

Methods

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.ToEigenVec	getToEigenVec()
hex.ModelMetrics.MetricBuilder	makeMetricBuilder(java.lang.String[] domain)
protected water.fvec.Frame	postProcessPredictions(water.fvec.Frame predictFr)
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)
water.fvec.Frame	scoreLeafNodeAssignment(water.fvec.Frame frame, 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, auc, checksum_impl, classification_error, cleanup_adapt, closeBigScorePredict, compareTo, computeDeviances, data, defaultThreshold, deleteCrossValidationModels, deviance, deviance, fillScoringInfo, getDefaultGridSortBy, getMojo, isSupervised, last_scored, lift_top_group, logloss, loss, mae, makeBigScoreTask, makeInteraction, makeInteractions, makeInteractions, makeSchema, makeScoringNames, mean_per_class_error, mse, predictScoreImpl, rmsle, score, score, score, score, score, score0, score0, scoreMetrics, scoring_history, scoringDomains, setupBigScorePredict, testJavaScoring, testJavaScoring, testJavaScoring, toJava, toJava, toJava, toJavaNCLASSES, toJavaPROB, toJavaSuper, 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>

scoreLeafNodeAssignment

public water.fvec.Frame scoreLeafNodeAssignment(water.fvec.Frame frame,
                                       water.Key<water.fvec.Frame> destination_key)

Specified by:

scoreLeafNodeAssignment in interface hex.Model.LeafNodeAssignment

postProcessPredictions

protected water.fvec.Frame postProcessPredictions(water.fvec.Frame predictFr)

Overrides:

postProcessPredictions 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)

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)

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>

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)