hex.gbm

Class DTree.TreeModel

java.lang.Object

water.Iced

water.Lockable<Model>

water.Model

hex.gbm.DTree.TreeModel

All Implemented Interfaces:

java.lang.Cloneable, Freezable

Direct Known Subclasses:

DRF.DRFModel, GBM.GBMModel

Enclosing class:

DTree

public abstract static class DTree.TreeModel
extends Model

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	DTree.TreeModel.CompressedTree
static class	DTree.TreeModel.TreeStats
static class	DTree.TreeModel.TreeVisitor<T extends java.lang.Exception> Abstract visitor class for serialized trees.

Field Summary

Fields

Modifier and Type	Field and Description
ConfusionMatrix[]	cms
static DocGen.FieldDoc[]	DOC_FIELDS
double[]	errs
int	max_depth
int	min_rows
int	N
int	nbins
Key	testKey
DTree.TreeModel.CompressedTree[][]	treeBits
DTree.TreeModel.TreeStats	treeStats
float[]	varimp

Fields inherited from class water.Model

_dataKey, _domains, _names

Fields inherited from class water.Lockable

_key, _lockers

Constructor Summary

Constructors

Constructor and Description
DTree.TreeModel(DTree.TreeModel prior, double err, ConfusionMatrix cm)
DTree.TreeModel(DTree.TreeModel prior, DTree[] trees, double err, ConfusionMatrix cm, DTree.TreeModel.TreeStats tstats)
DTree.TreeModel(DTree.TreeModel prior, DTree[] trees, DTree.TreeModel.TreeStats tstats)
DTree.TreeModel(DTree.TreeModel prior, float[] varimp)
DTree.TreeModel(Key key, Key dataKey, Key testKey, java.lang.String[] names, java.lang.String[][] domains, int ntrees, int max_depth, int min_rows, int nbins)

Method Summary

Methods

Modifier and Type	Method and Description
ConfusionMatrix	cm() For classifiers, confusion matrix on validation set.
void	generateHTML(java.lang.String title, java.lang.StringBuilder sb)
protected void	generateHTMLTreeStats(java.lang.StringBuilder sb)
protected void	generateHTMLVarImp(java.lang.StringBuilder sb)
protected abstract void	generateModelDescription(java.lang.StringBuilder sb)
double	mse() Returns mse for validation set.
int	numTrees()
protected float[]	score0(double[] data, float[] preds) Subclasses implement the scoring logic.
void	score0(double[] data, float[] preds, int treeIdx)
protected java.lang.String	toJavaDefaultMaxIters()
protected void	toJavaFillPreds0(SB bodySb) Fill preds[0] based on already filled and unified preds[1,..NCLASSES].
void	toJavaHtml(java.lang.StringBuilder sb)
protected SB	toJavaInit(SB sb, SB fileContextSB)
protected void	toJavaPredictBody(SB bodySb, SB classCtxSb, SB fileCtxSb)
protected void	toJavaTreePredictFct(SB sb, DTree.TreeModel.CompressedTree cts, int treeIdx, int classIdx)
protected void	toJavaUnifyPreds(SB bodySb) Generates code which unify preds[1,...NCLASSES]
VariableImportance	varimp() Variable importance of individual variables measured by this model.

Methods inherited from class water.Model

adapt, classNames, delete_impl, errStr, getDomainMapping, getDomainMapping, getPrediction, getPrediction, isClassifier, nclasses, responseName, score, score, score, score, score, score0, 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

N

public final int N

errs

public final double[] errs

treeBits

public final DTree.TreeModel.CompressedTree[][] treeBits

max_depth

public final int max_depth

min_rows

public final int min_rows

nbins

public final int nbins

testKey

public final Key testKey

cms

public final ConfusionMatrix[] cms

varimp

public final float[] varimp

treeStats

public final DTree.TreeModel.TreeStats treeStats

Constructor Detail

DTree.TreeModel

public DTree.TreeModel(Key key,
               Key dataKey,
               Key testKey,
               java.lang.String[] names,
               java.lang.String[][] domains,
               int ntrees,
               int max_depth,
               int min_rows,
               int nbins)

DTree.TreeModel

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

DTree.TreeModel

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

DTree.TreeModel

public DTree.TreeModel(DTree.TreeModel prior,
               double err,
               ConfusionMatrix cm)

DTree.TreeModel

public DTree.TreeModel(DTree.TreeModel prior,
               float[] varimp)

Method Detail

numTrees

public int numTrees()

cm

public ConfusionMatrix cm()

Description copied from class: Model

For classifiers, confusion matrix on validation set.

Overrides:

cm in class Model

varimp

public VariableImportance varimp()

Description copied from class: Model

Variable importance of individual variables measured by this model.

Overrides:

varimp in class Model

mse

public double mse()

Description copied from class: Model

Returns mse for validation set.

Overrides:

mse in class Model

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.

Specified by:

score0 in class Model

score0

public void score0(double[] data,
          float[] preds,
          int treeIdx)

generateHTML

public void generateHTML(java.lang.String title,
                java.lang.StringBuilder sb)

generateHTMLTreeStats

protected void generateHTMLTreeStats(java.lang.StringBuilder sb)

generateHTMLVarImp

protected void generateHTMLVarImp(java.lang.StringBuilder sb)

generateModelDescription

protected abstract void generateModelDescription(java.lang.StringBuilder sb)

toJavaHtml

public void toJavaHtml(java.lang.StringBuilder sb)

toJavaInit

protected SB toJavaInit(SB sb,
            SB fileContextSB)

Overrides:

toJavaInit in class Model

toJavaPredictBody

protected void toJavaPredictBody(SB bodySb,
                     SB classCtxSb,
                     SB fileCtxSb)

Overrides:

toJavaPredictBody in class Model

toJavaUnifyPreds

protected void toJavaUnifyPreds(SB bodySb)

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

toJavaFillPreds0

protected void toJavaFillPreds0(SB bodySb)

Fill preds[0] based on already filled and unified preds[1,..NCLASSES].

toJavaTreePredictFct

protected void toJavaTreePredictFct(SB sb,
                        DTree.TreeModel.CompressedTree cts,
                        int treeIdx,
                        int classIdx)

toJavaDefaultMaxIters

protected java.lang.String toJavaDefaultMaxIters()

Overrides:

toJavaDefaultMaxIters in class Model