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

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.SB

Field Summary

long[][]	cm
static DocGen.FieldDoc[]	DOC_FIELDS
double[]	errs
int	N
Key	testKey
DTree.TreeModel.CompressedTree[][]	treeBits
DTree.TreeModel.TreeStats	treeStats
float[]	varimp

Fields inherited from class water.Model

_dataKey, _domains, _names, _selfKey

Constructor Summary

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)

Method Summary

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)
int	numTrees()
protected float[]	score0(double[] data, float[] preds) Subclasses implement the scoring logic.
void	score0(double[] data, float[] preds, int treeIdx)
protected void	toJavaPredictBody(Model.SB sb)
VariableImportance	varimp() Variable importance of individual variables measured by this model.

Methods inherited from class water.Model

adapt, adapt, classNames, delete, getDomainMapping, isClassifier, nclasses, responseName, score, score, score, score, score, score0, testJavaScoring, toJava, toJavaInit, 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

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)

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)

toJavaPredictBody

protected void toJavaPredictBody(Model.SB sb)

Overrides:

toJavaPredictBody in class Model