hex.tree

Class SharedTreeModel.SharedTreeOutput

java.lang.Object

water.Iced

hex.Model.Output

hex.tree.SharedTreeModel.SharedTreeOutput

All Implemented Interfaces:

hex.Model.GetNTrees, CalibrationHelper.OutputWithCalibration, java.io.Externalizable, java.io.Serializable, java.lang.Cloneable, water.Freezable

Direct Known Subclasses:

DRFModel.DRFOutput, GBMModel.GBMOutput, IsolationForestModel.IsolationForestOutput, UpliftDRFModel.UpliftDRFOutput

Enclosing class:

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

public abstract static class SharedTreeModel.SharedTreeOutput
extends hex.Model.Output
implements hex.Model.GetNTrees, CalibrationHelper.OutputWithCalibration

See Also:

Serialized Form

Field Summary

Fields

Modifier and Type	Field and Description
hex.Model<?,?,?>	_calib_model
double	_init_f InitF value (for zero trees) f0 = mean(yi) for gaussian f0 = log(yi/1-yi) for bernoulli For GBM bernoulli, the initial prediction for 0 trees is p = 1/(1+exp(-f0)) From this, the mse for 0 trees (null model) can be computed as follows: mean((yi-p)^2)
int	_ntrees Number of trees actually in the model (as opposed to requested)
hex.ScoreKeeper[]	_scored_train
hex.ScoreKeeper[]	_scored_valid
long[]	_training_time_ms Training time
water.Key<CompressedTree>[][]	_treeKeys Trees get big, so store each one separately in the DKV.
water.Key<CompressedTree>[][]	_treeKeysAux
TreeStats	_treeStats More indepth tree stats
water.util.TwoDimTable	_variable_importances Variable importances computed during training
hex.VarImp	_varimp

Fields inherited from class hex.Model.Output

_column_types, _cross_validation_fold_assignment_frame_id, _cross_validation_holdout_predictions_frame_id, _cross_validation_metrics, _cross_validation_metrics_summary, _cross_validation_models, _cross_validation_predictions, _cv_scoring_history, _defaultThreshold, _distribution, _domains, _end_time, _hasFold, _hasOffset, _hasTreatment, _hasWeights, _isSupervised, _job, _model_summary, _modelClassDist, _names, _orig_projection_array, _origDomains, _origNames, _priorClassDist, _reproducibility_information_table, _run_time, _scoring_history, _start_time, _total_run_time, _training_metrics, _validation_metrics

Constructor Summary

Constructors

Constructor and Description
SharedTreeOutput(SharedTree b)

Method Summary

Modifier and Type	Method and Description
void	addKTrees(DTree[] trees)
hex.Model<?,?,?>	calibrationModel()
water.util.TwoDimTable	createInputFramesInformationTable(hex.ModelBuilder modelBuilder)
CompressedTree	ctree(int tnum, int knum)
int	getInformationTableNumRows()
int	getNTrees()
water.util.TwoDimTable	getVariableImportances()
hex.ScoreKeeper[]	scoreKeepers()
void	setCalibrationModel(hex.Model<?,?,?> model)
java.lang.String	toStringTree(int tnum, int knum)
void	trimTo(int ntrees)

Methods inherited from class hex.Model.Output

changeModelMetricsKey, checksum_impl, classNames, clearModelMetrics, defaultThreshold, features, foldIdx, foldName, getModelCategory, getModelMetrics, hasFold, hasOffset, hasResponse, hasTreatment, hasWeights, interactionBuilder, isAutoencoder, isBinomialClassifier, isClassifier, isMultinomialClassifier, isSupervised, lastSpecialColumnIdx, nclasses, nfeatures, offsetIdx, offsetName, printTwoDimTables, resetThreshold, responseIdx, responseName, setNames, setNames, startClock, stopClock, toString, treatmentIdx, treatmentName, weightsIdx, weightsName

Methods inherited from class water.Iced

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

Methods inherited from class java.lang.Object

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

Methods inherited from interface hex.tree.CalibrationHelper.OutputWithCalibration

getCalibrationMethod, getModelCategory, isCalibrated

Field Detail

_init_f

public double _init_f

InitF value (for zero trees) f0 = mean(yi) for gaussian f0 = log(yi/1-yi) for bernoulli For GBM bernoulli, the initial prediction for 0 trees is p = 1/(1+exp(-f0)) From this, the mse for 0 trees (null model) can be computed as follows: mean((yi-p)^2)

_ntrees

public int _ntrees

Number of trees actually in the model (as opposed to requested)

_treeStats

public final TreeStats _treeStats

More indepth tree stats

_treeKeys

public water.Key<CompressedTree>[][] _treeKeys

Trees get big, so store each one separately in the DKV.

_treeKeysAux

public water.Key<CompressedTree>[][] _treeKeysAux

_scored_train

public hex.ScoreKeeper[] _scored_train

_scored_valid

public hex.ScoreKeeper[] _scored_valid

_training_time_ms

public long[] _training_time_ms

Training time

_variable_importances

public water.util.TwoDimTable _variable_importances

Variable importances computed during training

_varimp

public hex.VarImp _varimp

_calib_model

public hex.Model<?,?,?> _calib_model

Constructor Detail

SharedTreeOutput

public SharedTreeOutput(SharedTree b)

Method Detail

scoreKeepers

public hex.ScoreKeeper[] scoreKeepers()

getVariableImportances

public water.util.TwoDimTable getVariableImportances()

Overrides:

getVariableImportances in class hex.Model.Output

createInputFramesInformationTable

public water.util.TwoDimTable createInputFramesInformationTable(hex.ModelBuilder modelBuilder)

Overrides:

createInputFramesInformationTable in class hex.Model.Output

getInformationTableNumRows

public int getInformationTableNumRows()

Overrides:

getInformationTableNumRows in class hex.Model.Output

addKTrees

public void addKTrees(DTree[] trees)

trimTo

public void trimTo(int ntrees)

getNTrees

public int getNTrees()

Specified by:

getNTrees in interface hex.Model.GetNTrees

calibrationModel

public hex.Model<?,?,?> calibrationModel()

Specified by:

calibrationModel in interface CalibrationHelper.OutputWithCalibration

setCalibrationModel

public void setCalibrationModel(hex.Model<?,?,?> model)

Specified by:

setCalibrationModel in interface CalibrationHelper.OutputWithCalibration

ctree

public CompressedTree ctree(int tnum,
                            int knum)

toStringTree

public java.lang.String toStringTree(int tnum,
                                     int knum)