hex.gbm

Class GBM.GBMModel

java.lang.Object

water.Iced

water.Lockable<Model>

water.Model

hex.gbm.DTree.TreeModel

hex.gbm.GBM.GBMModel

All Implemented Interfaces:

java.lang.Cloneable, Freezable

Enclosing class:

GBM

public static class GBM.GBMModel
extends DTree.TreeModel

Nested Class Summary

Nested classes/interfaces inherited from class hex.gbm.DTree.TreeModel

DTree.TreeModel.CompressedTree, DTree.TreeModel.TreeModelType, DTree.TreeModel.TreeStats, DTree.TreeModel.TreeVisitor<T extends java.lang.Exception>

Nested classes/interfaces inherited from class water.Model

Model.ModelAutobufferSerializer, Model.ModelCategory

Field Summary

Fields

Modifier and Type	Field and Description
static DocGen.FieldDoc[]	DOC_FIELDS

Fields inherited from class hex.gbm.DTree.TreeModel

cmDomain, cms, errs, isFromSpeeDRF, max_depth, min_rows, N, nbins, testKey, treeKeys, treeStats, validAUC, varimp

Fields inherited from class water.Model

_dataKey, _domains, _have_cv_results, _modelClassDist, _names, _priorClassDist, GEN_BENCHMARK_CODE, training_duration_in_ms, training_start_time, warnings

Fields inherited from class water.Lockable

_key, _lockers

Constructor Summary

Constructors

Constructor and Description
GBM.GBMModel(GBM job, Key key, Key dataKey, Key testKey, java.lang.String[] names, java.lang.String[][] domains, java.lang.String[] cmDomain, int ntrees, int max_depth, int min_rows, int nbins, double learn_rate, GBM.Family family, int num_folds, float[] priorClassDist, float[] classDist)

Method Summary

Methods

Modifier and Type	Method and Description
protected void	generateModelDescription(java.lang.StringBuilder sb)
GBM	get_params()
protected DTree.TreeModel.TreeModelType	getTreeModelType()
Request2	job()
protected float[]	score0(double[] data, float[] preds) Subclasses implement the scoring logic.
protected void	setCrossValidationError(Job.ValidatedJob job, double cv_error, ConfusionMatrix cm, AUCData auc, HitRatio hr)
protected void	toJavaUnifyPreds(SB bodyCtxSB) Generates code which unify preds[1,...NCLASSES]

Methods inherited from class hex.gbm.DTree.TreeModel

cm, ctree, delete_impl, delete_trees, delete_trees, generateHTML, generateHTMLAUC, generateHTMLTreeStats, generateHTMLVarImp, getModelSerializer, mse, ntrees, score0, toJavaDefaultMaxIters, toJavaHtml, toJavaInit, toJavaPredictBody, toJavaTreePredictFct, varimp

Methods inherited from class water.Model

adapt, adapt, addWarning, calcError, classNames, errStr, getDomainMapping, getDomainMapping, getModelCategory, getUniqueId, hasCrossValModels, isClassifier, isSupervised, missingColumnsType, nclasses, nfeatures, printCrossValidationModelsHTML, responseName, score, score, score, score, score, score, score0, scoreCrossValidation, scoreImpl, setModelClassDistribution, start_training, start_training, stop_training, testJavaScoring, toJava, toJava, toJavaFillPreds0, toJavaInit, toJavaNCLASSES, toJavaSuper

Methods inherited from class water.Lockable

delete_and_lock, delete, delete, delete, delete, is_unlocked, is_wlocked, read_lock, read_lock, unlock_all, unlock_lockable, unlock, update, write_lock

Methods inherited from class water.Iced

clone, frozenType, init, newInstance, read, toDocField, write, writeJSON, writeJSONFields

Methods inherited from class java.lang.Object

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

Field Detail

DOC_FIELDS

public static DocGen.FieldDoc[] DOC_FIELDS

Constructor Detail

GBM.GBMModel

public GBM.GBMModel(GBM job,
            Key key,
            Key dataKey,
            Key testKey,
            java.lang.String[] names,
            java.lang.String[][] domains,
            java.lang.String[] cmDomain,
            int ntrees,
            int max_depth,
            int min_rows,
            int nbins,
            double learn_rate,
            GBM.Family family,
            int num_folds,
            float[] priorClassDist,
            float[] classDist)

Method Detail

get_params

public final GBM get_params()

Overrides:

get_params in class Model

job

public final Request2 job()

Overrides:

job in class Model

getTreeModelType

protected DTree.TreeModel.TreeModelType getTreeModelType()

Overrides:

getTreeModelType in class DTree.TreeModel

score0

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

Description copied from class: Model

Subclasses implement the scoring logic. The data is pre-loaded into a re-used temp array, in the order the model expects. The predictions are loaded into the re-used temp array, which is also returned.

Overrides:

score0 in class DTree.TreeModel

generateModelDescription

protected void generateModelDescription(java.lang.StringBuilder sb)

Specified by:

generateModelDescription in class DTree.TreeModel

toJavaUnifyPreds

protected void toJavaUnifyPreds(SB bodyCtxSB)

Description copied from class: Model

Generates code which unify preds[1,...NCLASSES]

Overrides:

toJavaUnifyPreds in class Model

setCrossValidationError

protected void setCrossValidationError(Job.ValidatedJob job,
                           double cv_error,
                           ConfusionMatrix cm,
                           AUCData auc,
                           HitRatio hr)

Overrides:

setCrossValidationError in class Model