hex.gbm

Class DTree.TreeModel

java.lang.Object

water.Iced

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.

Nested classes/interfaces inherited from class water.Model

Model.GeneratedModel

Field Summary

Fields

Modifier and Type	Field and Description
long[][]	cm
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, _selfKey

Constructor Summary

Constructors

Constructor and Description
DTree.TreeModel(DTree.TreeModel prior, DTree[] trees, double err, long[][] cm, 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)
int	numTrees()
protected float[]	score0(double[] data, float[] preds) Subclasses implement the scoring logic.
void	score0(double[] data, float[] preds, int treeIdx)
void	toJavaHtml(java.lang.StringBuilder sb)
protected SB	toJavaInit(SB sb)
protected void	toJavaPredictBody(SB sb, SB afterBodySb)
protected void	toJavaTreePredictFct(SB sb, DTree.TreeModel.CompressedTree cts, int tidx, int c)
VariableImportance	varimp() Variable importance of individual variables measured by this model.

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

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

cm

public final long[][] cm

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,
               float[] varimp)

DTree.TreeModel

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

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

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)

Overrides:

toJavaInit in class Model

toJavaPredictBody

protected void toJavaPredictBody(SB sb,
                     SB afterBodySb)

Overrides:

toJavaPredictBody in class Model

toJavaTreePredictFct

protected void toJavaTreePredictFct(SB sb,
                        DTree.TreeModel.CompressedTree cts,
                        int tidx,
                        int c)