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
java.lang.String[]	cmDomain
ConfusionMatrix[]	cms
static DocGen.FieldDoc[]	DOC_FIELDS
double[]	errs
int	max_depth
int	min_rows
int	N
int	nbins
Key	testKey
Key[][]	treeKeys
DTree.TreeModel.TreeStats	treeStats
AUC	validAUC
VarImp	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
DTree.TreeModel(DTree.TreeModel prior, double err, ConfusionMatrix cm, VarImp varimp, AUC validAUC)
DTree.TreeModel(DTree.TreeModel prior, DTree[] tree, double err, ConfusionMatrix cm, DTree.TreeModel.TreeStats tstats)
DTree.TreeModel(DTree.TreeModel prior, DTree[] tree, DTree.TreeModel.TreeStats tstats)
DTree.TreeModel(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)

Method Summary

Methods

Modifier and Type	Method and Description
ConfusionMatrix	cm() For classifiers, confusion matrix on validation set.
DTree.TreeModel.CompressedTree[]	ctree(int tidx) Returns i-th tree represented by an array of k-trees.
Futures	delete_impl(Futures fs) Remove any Model internal Keys
void	delete_trees() Delete model trees
Futures	delete_trees(Futures fs)
void	generateHTML(java.lang.String title, java.lang.StringBuilder sb)
protected void	generateHTMLAUC(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	ntrees() Number of trees in current model.
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]
VarImp	varimp() Variable importance of individual input features measured by this model.

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

N

public final int N

errs

public final double[] errs

treeKeys

public final Key[][] treeKeys

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

cmDomain

public final java.lang.String[] cmDomain

varimp

public final VarImp varimp

treeStats

public final DTree.TreeModel.TreeStats treeStats

validAUC

public final AUC validAUC

Constructor Detail

DTree.TreeModel

public DTree.TreeModel(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)

DTree.TreeModel

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

DTree.TreeModel

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

DTree.TreeModel

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

Method Detail

ntrees

public int ntrees()

Number of trees in current model.

cm

public ConfusionMatrix cm()

Description copied from class: Model

For classifiers, confusion matrix on validation set.

Overrides:

cm in class Model

varimp

public VarImp varimp()

Description copied from class: Model

Variable importance of individual input features 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

ctree

public final DTree.TreeModel.CompressedTree[] ctree(int tidx)

Returns i-th tree represented by an array of k-trees.

score0

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

delete_trees

public void delete_trees()

Delete model trees

delete_trees

public Futures delete_trees(Futures fs)

delete_impl

public Futures delete_impl(Futures fs)

Description copied from class: Model

Remove any Model internal Keys

Overrides:

delete_impl in class Model

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)

generateHTMLAUC

protected void generateHTMLAUC(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