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

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.DeepFeatures, hex.Model.ExemplarMembers, hex.Model.GLRMArchetypes, 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 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.
hex.ModelMetrics.MetricBuilder	makeMetricBuilder(java.lang.String[] domain)
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 weight, double offset)
protected double[]	score0(double[] data, double[] preds, double weight, double offset, int ntrees)
water.fvec.Frame	scoreLeafNodeAssignment(water.fvec.Frame frame, water.Key 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, addWarning, auc, checksum_impl, classification_error, cleanup_adapt, compareTo, defaultThreshold, deleteCrossValidationModels, deviance, deviance, fillScoringInfo, getToEigenVec, isSupervised, last_scored, lift_top_group, logloss, loss, mae, makeInteraction, makeInteractions, makeInteractions, makeSchema, makeScoringNames, mean_per_class_error, mse, predictScoreImpl, rmsle, score, score, score, score, score0, score0, scoreMetrics, scoring_history, scoringDomains, 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 selfKey,
               P parms,
               O output)

Method Detail

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

Specified by:

scoreLeafNodeAssignment in interface hex.Model.LeafNodeAssignment

score0

protected double[] score0(double[] data,
              double[] preds,
              double weight,
              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 weight,
              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)