hex.gbm

Class GBM.GBMModel

java.lang.Object

water.Iced

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>

Nested classes/interfaces inherited from class water.Model

Model.GeneratedModel

Field Summary

Fields

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

Fields inherited from class hex.gbm.DTree.TreeModel

cm, errs, max_depth, min_rows, N, nbins, testKey, treeBits, treeStats, varimp

Fields inherited from class water.Model

_dataKey, _domains, _names, _selfKey

Constructor Summary

Constructors

Constructor and Description
GBM.GBMModel(GBM.GBMModel prior, DTree[] trees, double err, long[][] cm, DTree.TreeModel.TreeStats tstats)
GBM.GBMModel(Key key, Key dataKey, Key testKey, java.lang.String[] names, java.lang.String[][] domains, 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	toJavaPredictBody(SB sb, SB afterBodySb)

Methods inherited from class hex.gbm.DTree.TreeModel

cm, generateHTML, generateHTMLTreeStats, generateHTMLVarImp, numTrees, score0, toJavaHtml, toJavaInit, toJavaTreePredictFct, varimp

Methods inherited from class water.Model

adapt, classNames, delete, getDomainMapping, isClassifier, nclasses, responseName, score, score, score, score, score, score0, testJavaScoring, toJava, toJava, toJavaInit

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,
            int ntrees,
            int max_depth,
            int min_rows,
            int nbins,
            double learn_rate)

GBM.GBMModel

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

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

toJavaPredictBody

protected void toJavaPredictBody(SB sb,
                     SB afterBodySb)

Overrides:

toJavaPredictBody in class DTree.TreeModel