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.TreeStats, DTree.TreeModel.TreeVisitor<T extends java.lang.Exception>

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, max_depth, min_rows, N, nbins, testKey, treeKeys, treeStats, validAUC, varimp

Fields inherited from class water.Model

_dataKey, _domains, _modelClassDist, _names, _priorClassDist

Fields inherited from class water.Lockable

_key, _lockers

Constructor Summary

Constructors

Constructor and Description
GBM.GBMModel(DTree.TreeModel prior, double err, ConfusionMatrix cm, VarImp varimp, AUC validAUC)
GBM.GBMModel(DTree.TreeModel prior, DTree[] trees, double err, ConfusionMatrix cm, DTree.TreeModel.TreeStats tstats)
GBM.GBMModel(DTree.TreeModel prior, DTree[] trees, DTree.TreeModel.TreeStats tstats)
GBM.GBMModel(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)

Method Summary

Methods

Modifier and Type	Method and Description
protected void	generateModelDescription(java.lang.StringBuilder sb)
protected float[]	score0(double[] data, float[] preds) Subclasses implement the scoring logic.
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, mse, ntrees, score0, toJavaDefaultMaxIters, toJavaFillPreds0, toJavaHtml, toJavaInit, toJavaPredictBody, toJavaTreePredictFct, varimp

Methods inherited from class water.Model

adapt, classNames, errStr, getDomainMapping, getDomainMapping, isClassifier, nclasses, nfeatures, responseName, score, score, score, score, score, score, score0, setModelClassDistribution, testJavaScoring, toJava, toJava, toJavaInit, toJavaSuper

Methods inherited from class water.Lockable

delete_and_lock, delete, delete, delete, delete, read_lock, read_lock, 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(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.GBMModel

public GBM.GBMModel(DTree.TreeModel prior,
            DTree[] trees,
            double err,
            ConfusionMatrix cm,
            DTree.TreeModel.TreeStats tstats)

GBM.GBMModel

public GBM.GBMModel(DTree.TreeModel prior,
            DTree[] trees,
            DTree.TreeModel.TreeStats tstats)

GBM.GBMModel

public GBM.GBMModel(DTree.TreeModel prior,
            double err,
            ConfusionMatrix cm,
            VarImp varimp,
            AUC validAUC)

Method Detail

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: DTree.TreeModel

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

Overrides:

toJavaUnifyPreds in class DTree.TreeModel