hex.drf

Class DRF.DRFModel

java.lang.Object

water.Iced

water.Lockable<Model>

water.Model

hex.gbm.DTree.TreeModel

hex.drf.DRF.DRFModel

All Implemented Interfaces:

java.lang.Cloneable, Freezable

Enclosing class:

DRF

public static class DRF.DRFModel
extends DTree.TreeModel

DRF model holding serialized tree and implementing logic for scoring a row

Nested Class Summary

Nested classes/interfaces inherited from class hex.gbm.DTree.TreeModel

DTree.TreeModel.CompressedTree, DTree.TreeModel.TreeModelType, DTree.TreeModel.TreeStats, DTree.TreeModel.TreeVisitor<T extends java.lang.Exception>

Nested classes/interfaces inherited from class water.Model

Model.ModelCategory

Field Summary

Fields

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

Fields inherited from class hex.gbm.DTree.TreeModel

cmDomain, cms, errs, max_depth, min_rows, N, nbins, testKey, treeKeys, treeStats, validAUC, varimp

Fields inherited from class water.Model

_dataKey, _domains, _modelClassDist, _names, _priorClassDist, training_duration_in_ms, training_start_time

Fields inherited from class water.Lockable

_key, _lockers

Constructor Summary

Constructors

Constructor and Description
DRF.DRFModel(DRF params, 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, int mtries, float sample_rate, long seed)

Method Summary

Methods

Modifier and Type	Method and Description
protected void	generateModelDescription(java.lang.StringBuilder sb)
DRF	get_params()
protected DTree.TreeModel.TreeModelType	getTreeModelType()
Request2	job()
protected float[]	score0(double[] data, float[] preds) Subclasses implement the scoring logic.
protected void	toJavaUnifyPreds(SB bodySb) Generates code which unify preds[1,...NCLASSES]

Methods inherited from class hex.gbm.DTree.TreeModel

cm, ctree, delete_impl, delete_trees, delete_trees, generateHTML, generateHTMLAUC, generateHTMLTreeStats, generateHTMLVarImp, mse, ntrees, score0, toJavaDefaultMaxIters, toJavaFillPreds0, toJavaHtml, toJavaInit, toJavaPredictBody, toJavaTreePredictFct, varimp

Methods inherited from class water.Model

adapt, calcError, classNames, errStr, getDomainMapping, getDomainMapping, getModelCategory, getUniqueId, isClassifier, missingColumnsType, nclasses, nfeatures, responseName, score, score, score, score, score, score, score0, setModelClassDistribution, start_training, start_training, stop_training, testJavaScoring, toJava, toJava, toJavaInit, toJavaSuper

Methods inherited from class water.Lockable

delete_and_lock, delete, delete, delete, delete, is_unlocked, is_wlocked, read_lock, read_lock, unlock_all, 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

Constructor Detail

DRF.DRFModel

public DRF.DRFModel(DRF params,
            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,
            int mtries,
            float sample_rate,
            long seed)

Method Detail

get_params

public final DRF get_params()

Overrides:

get_params in class Model

job

public final Request2 job()

Overrides:

job in class Model

getTreeModelType

protected DTree.TreeModel.TreeModelType getTreeModelType()

Overrides:

getTreeModelType in class DTree.TreeModel

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

toJavaUnifyPreds

protected void toJavaUnifyPreds(SB bodySb)

Description copied from class: DTree.TreeModel

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

Overrides:

toJavaUnifyPreds in class DTree.TreeModel