hex.tree.gbm

Class GBMModel

java.lang.Object

water.Iced<T>

water.Keyed<T>

water.Lockable<M>

hex.Model<M,P,O>

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

hex.tree.SharedTreeModelWithContributions<GBMModel,GBMModel.GBMParameters,GBMModel.GBMOutput>

hex.tree.gbm.GBMModel

All Implemented Interfaces:

hex.FeatureInteractionsCollector, hex.FriedmanPopescusHCollector, hex.Model.Contributions, hex.Model.FeatureFrequencies, hex.Model.GetMostImportantFeatures, hex.Model.LeafNodeAssignment, hex.Model.RowToTreeAssignment, hex.Model.StagedPredictions, hex.Model.UpdateAuxTreeWeights, java.io.Externalizable, java.io.Serializable, java.lang.Cloneable, water.api.StreamWriter, water.Freezable<GBMModel>

public class GBMModel
extends SharedTreeModelWithContributions<GBMModel,GBMModel.GBMParameters,GBMModel.GBMOutput>
implements hex.Model.StagedPredictions, hex.FeatureInteractionsCollector, hex.FriedmanPopescusHCollector, hex.Model.RowToTreeAssignment

See Also:

Serialized Form

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	GBMModel.GBMOutput
static class	GBMModel.GBMParameters

Nested classes/interfaces inherited from class hex.tree.SharedTreeModelWithContributions

SharedTreeModelWithContributions.ScoreContributionsSortingTask, SharedTreeModelWithContributions.ScoreContributionsTask, SharedTreeModelWithContributions.ScoreContributionsWithBackgroundTask

Nested classes/interfaces inherited from class hex.tree.SharedTreeModel

SharedTreeModel.BufStringDecisionPathTracker, SharedTreeModel.SharedTreeOutput, SharedTreeModel.SharedTreeParameters

Nested classes/interfaces inherited from class hex.Model

hex.Model.AdaptFrameParameters, hex.Model.BigScore, hex.Model.BigScoreChunkPredict, hex.Model.BigScorePredict, hex.Model.Contributions, hex.Model.DeepFeatures, hex.Model.ExemplarMembers, hex.Model.FeatureFrequencies, hex.Model.GetMostImportantFeatures, hex.Model.GetNTrees, hex.Model.GLRMArchetypes, hex.Model.GridSortBy, hex.Model.H2OModelDescriptor, hex.Model.InteractionBuilder, hex.Model.InteractionPair, hex.Model.InteractionSpec, hex.Model.JavaModelStreamWriter, hex.Model.JavaScoringOptions, hex.Model.LeafNodeAssignment, hex.Model.Output, hex.Model.Parameters, hex.Model.PredictScoreResult, hex.Model.RowToTreeAssignment, hex.Model.StagedPredictions, hex.Model.UpdateAuxTreeWeights

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

hex.Model.Contributions.ContributionsOptions, hex.Model.Contributions.ContributionsOutputFormat

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

hex.Model.LeafNodeAssignment.LeafNodeAssignmentType

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

hex.Model.UpdateAuxTreeWeights.UpdateAuxTreeWeightsReport

Field Summary

Fields inherited from class hex.Model

_dist, _input_parms, _output, _parms, _toDelete, _warnings, _warningsP, EVAL_AUTO_PARAMS_ENABLED, evalAutoParamsEnabled, scoringInfo

Fields inherited from class water.Lockable

_lockers

Fields inherited from class water.Keyed

_key

Constructor Summary

Constructors

Constructor and Description
GBMModel(water.Key<GBMModel> selfKey, GBMModel.GBMParameters parms, GBMModel.GBMOutput output)

Method Summary

Modifier and Type	Method and Description
hex.FeatureInteractions	getFeatureInteractions(int maxInteractionDepth, int maxTreeDepth, int maxDeepening)
water.util.TwoDimTable[][]	getFeatureInteractionsTable(int maxInteractionDepth, int maxTreeDepth, int maxDeepening)
double	getFriedmanPopescusH(water.fvec.Frame frame, java.lang.String[] vars)
GbmMojoWriter	getMojo()
protected SharedTreeModelWithContributions.ScoreContributionsTask	getScoreContributionsSoringTask(SharedTreeModel model, hex.Model.Contributions.ContributionsOptions options)
protected SharedTreeModelWithContributions.ScoreContributionsTask	getScoreContributionsTask(SharedTreeModel model)
protected SharedTreeModelWithContributions.ScoreContributionsWithBackgroundTask	getScoreContributionsWithBackgroundTask(SharedTreeModel model, water.fvec.Frame fr, water.fvec.Frame backgroundFrame, boolean expand, int[] catOffsets, hex.Model.Contributions.ContributionsOptions options)
void	initActualParamValues()
void	initActualParamValuesAfterOutputSetup(int nclasses, boolean isClassifier)
protected SharedTreePojoWriter	makeTreePojoWriter()
water.fvec.Frame	rowToTreeAssignment(water.fvec.Frame frame, water.Key<water.fvec.Frame> destination_key, water.Job<water.fvec.Frame> j)
double	score(double[] data)
protected double[]	score0(double[] data, double[] preds, double offset, int ntrees) Bulk scoring API for one row.
protected double[]	score0Incremental(Score.ScoreIncInfo sii, water.fvec.Chunk[] chks, double offset, int row_in_chunk, double[] tmp, double[] preds)
water.fvec.Frame	scoreStagedPredictions(water.fvec.Frame frame, water.Key<water.fvec.Frame> destination_key)

Methods inherited from class hex.tree.SharedTreeModelWithContributions

removeSpecialColumns, removeSpecialNNonNumericColumns, scoreContributions, scoreContributions, scoreContributions, scoreContributions

Methods inherited from class hex.tree.SharedTreeModel

binomialOpt, deepClone, getGenModelEncoding, getMostImportantFeatures, getSharedTreeSubgraph, getToEigenVec, isFeatureUsedInPredict, makeAllTreeColumnNames, makeMetricBuilder, makePojoWriter, postProcessPredictions, readAll_impl, remove_impl, score0, score0, score0, scoreFeatureFrequencies, scoreLeafNodeAssignment, updateAuxTreeWeights, writeAll_impl

Methods inherited from class hex.Model

adaptFrameForScore, adaptTestForJavaScoring, adaptTestForTrain, adaptTestForTrain, adaptTestForTrain, addMetrics, addModelMetrics, addWarning, aic, auc, AUCPR, checksum_impl, classification_error, compareTo, computeDeviances, containsResponse, data, defaultThreshold, defaultThreshold, deleteCrossValidationFoldAssignment, deleteCrossValidationModels, deleteCrossValidationPreds, deviance, deviance, evaluateAutoModelParameters, exportBinaryModel, exportMojo, fetchAll, fillScoringInfo, getDefaultGridSortBy, getPojoInterfaces, haveMojo, havePojo, importBinaryModel, isDistributionHuber, isFeatureUsedInPredict, isGeneric, isSupervised, last_scored, lift_top_group, likelihood, logloss, loss, mae, makeAdaptFrameParameters, makeBigScoreTask, makeInteraction, makeInteractions, makeInteractions, makeSchema, makeScoringDomains, makeScoringNames, makeScoringNames, mean_per_class_error, modelDescriptor, mse, needsPostProcess, predictScoreImpl, r2, resetThreshold, result, rmsle, score, score, score, score, score, score, score, score0, score0, score0PostProcessSupervised, scoreMetrics, scoring_history, scoringDomains, setInputParms, setupBigScorePredict, testJavaScoring, testJavaScoring, testJavaScoring, testJavaScoring, testJavaScoring, toJava, toJava, toJava, toJavaAlgo, toJavaCheckTooBig, toJavaInit, toJavaModelClassName, toJavaPredictBody, toJavaTransform, toJavaUUID, toMojo, toMojo, toString, transform, uploadBinaryModel, writeTo

Methods inherited from class water.Lockable

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

Methods inherited from class water.Keyed

checksum_impl, checksum, checksum, getKey, readAll, remove_impl, remove_self_key_impl, remove, remove, remove, remove, remove, remove, removeQuietly, writeAll

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

composeScoreContributionTaskMetadata, scoreContributionsWorkEstimate

Constructor Detail

GBMModel

public GBMModel(water.Key<GBMModel> selfKey,
                GBMModel.GBMParameters parms,
                GBMModel.GBMOutput output)

Method Detail

initActualParamValues

public void initActualParamValues()

Overrides:

initActualParamValues in class hex.Model<GBMModel,GBMModel.GBMParameters,GBMModel.GBMOutput>

initActualParamValuesAfterOutputSetup

public void initActualParamValuesAfterOutputSetup(int nclasses,
                                                  boolean isClassifier)

getScoreContributionsWithBackgroundTask

protected SharedTreeModelWithContributions.ScoreContributionsWithBackgroundTask getScoreContributionsWithBackgroundTask(SharedTreeModel model,
                                                                                                                        water.fvec.Frame fr,
                                                                                                                        water.fvec.Frame backgroundFrame,
                                                                                                                        boolean expand,
                                                                                                                        int[] catOffsets,
                                                                                                                        hex.Model.Contributions.ContributionsOptions options)

Specified by:

getScoreContributionsWithBackgroundTask in class SharedTreeModelWithContributions<GBMModel,GBMModel.GBMParameters,GBMModel.GBMOutput>

getScoreContributionsTask

protected SharedTreeModelWithContributions.ScoreContributionsTask getScoreContributionsTask(SharedTreeModel model)

Specified by:

getScoreContributionsTask in class SharedTreeModelWithContributions<GBMModel,GBMModel.GBMParameters,GBMModel.GBMOutput>

getScoreContributionsSoringTask

protected SharedTreeModelWithContributions.ScoreContributionsTask getScoreContributionsSoringTask(SharedTreeModel model,
                                                                                                  hex.Model.Contributions.ContributionsOptions options)

Specified by:

getScoreContributionsSoringTask in class SharedTreeModelWithContributions<GBMModel,GBMModel.GBMParameters,GBMModel.GBMOutput>

scoreStagedPredictions

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

Specified by:

scoreStagedPredictions in interface hex.Model.StagedPredictions

score0Incremental

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

Overrides:

score0Incremental in class SharedTreeModel<GBMModel,GBMModel.GBMParameters,GBMModel.GBMOutput>

score0

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

Bulk scoring API for one row. Chunks are all compatible with the model, and expect the last Chunks are for the final distribution and prediction. Default method is to just load the data into the tmp array, then call subclass scoring logic.

Overrides:

score0 in class SharedTreeModel<GBMModel,GBMModel.GBMParameters,GBMModel.GBMOutput>

makeTreePojoWriter

protected SharedTreePojoWriter makeTreePojoWriter()

Overrides:

makeTreePojoWriter in class SharedTreeModel<GBMModel,GBMModel.GBMParameters,GBMModel.GBMOutput>

getMojo

public GbmMojoWriter getMojo()

Overrides:

getMojo in class hex.Model<GBMModel,GBMModel.GBMParameters,GBMModel.GBMOutput>

getFeatureInteractions

public hex.FeatureInteractions getFeatureInteractions(int maxInteractionDepth,
                                                      int maxTreeDepth,
                                                      int maxDeepening)

getFeatureInteractionsTable

public water.util.TwoDimTable[][] getFeatureInteractionsTable(int maxInteractionDepth,
                                                              int maxTreeDepth,
                                                              int maxDeepening)

Specified by:

getFeatureInteractionsTable in interface hex.FeatureInteractionsCollector

getFriedmanPopescusH

public double getFriedmanPopescusH(water.fvec.Frame frame,
                                   java.lang.String[] vars)

Specified by:

getFriedmanPopescusH in interface hex.FriedmanPopescusHCollector

rowToTreeAssignment

public water.fvec.Frame rowToTreeAssignment(water.fvec.Frame frame,
                                            water.Key<water.fvec.Frame> destination_key,
                                            water.Job<water.fvec.Frame> j)

Specified by:

rowToTreeAssignment in interface hex.Model.RowToTreeAssignment

score

public double score(double[] data)

Overrides:

score in class hex.Model<GBMModel,GBMModel.GBMParameters,GBMModel.GBMOutput>