hex.singlenoderf

Class SpeeDRFModel

java.lang.Object

water.Iced

water.Lockable<Model>

water.Model

hex.singlenoderf.SpeeDRFModel

All Implemented Interfaces:

java.lang.Cloneable, Freezable, Job.Progress

public class SpeeDRFModel
extends Model
implements Job.Progress

Nested Class Summary

Nested classes/interfaces inherited from class water.Model

Model.ModelCategory

Field Summary

Fields

Modifier and Type	Field and Description
static DocGen.FieldDoc[]	DOC_FIELDS
static java.lang.String	JSON_CM_CLASS_ERR
static java.lang.String	JSON_CM_CLASSES_ERRORS
static java.lang.String	JSON_CM_HEADER
static java.lang.String	JSON_CM_MATRIX
static java.lang.String	JSON_CM_ROWS
static java.lang.String	JSON_CM_ROWS_SKIPPED
static java.lang.String	JSON_CM_TREES
static java.lang.String	JSON_CM_TYPE
static java.lang.String	JSON_CONFUSION_KEY
AUC	validAUC
VarImp	varimp
protected long	zeed

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
SpeeDRFModel(Key selfKey, Key jobKey, Key dataKey, Frame fr, Vec response, Key[] t_keys, long zeed, java.lang.String[] cmDomain, SpeeDRF params)

Method Summary

Methods

Modifier and Type	Method and Description
void	buildCM(java.lang.StringBuilder sb)
int	classes()
short	classify(Frame fr, Chunk[] chks, int row, int[] modelDataMap, int[] votes, double[] classWt, java.util.Random rand)
short	classify(int[] votes, double[] classWt, java.util.Random rand)
float	classify0(int tree_id, Frame fr, Chunk[] chunks, int row, int[] modelDataMap, short badrow, boolean regression) Classify a row according to one particular tree.
int[]	colMap(java.lang.String[] names)
static float[]	computeVarImpSD(float[][] vote_diffs)
Futures	delete_impl(Futures fs) Free all internal tree keys.
Counter	depth()
protected VarImp	doVarImpCalc(SpeeDRFModel model)
void	find_leaves_depth() Internal computation of depth and number of leaves.
void	generateHTML(java.lang.String title, java.lang.StringBuilder sb)
protected void	generateHTMLAUC(java.lang.StringBuilder sb)
void	generateHTMLTreeStats(java.lang.StringBuilder sb, com.google.gson.JsonObject trees)
protected void	generateHTMLVarImp(java.lang.StringBuilder sb)
SpeeDRF	get_params()
Vec	get_response()
Request2	job()
Counter	leaves()
static SpeeDRFModel	make(SpeeDRFModel old, Key tkey, int nodeIdx)
protected static AUC	makeAUC(ConfusionMatrix[] cms, float[] threshold, java.lang.String[] cmDomain)
java.lang.String	name(int atree)
float	progress()
protected float[]	score0(double[] data, float[] preds) Subclasses implement the scoring logic.
int	size()
byte[]	tree(int tree_id) Return the bits for a particular tree
int	treeCount()

Methods inherited from class water.Model

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

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

zeed

protected long zeed

validAUC

@Request.API(help="AUC")
public AUC validAUC

varimp

@Request.API(help="Variable Importance")
public VarImp varimp

JSON_CONFUSION_KEY

public static final java.lang.String JSON_CONFUSION_KEY

See Also:

Constant Field Values

JSON_CM_TYPE

public static final java.lang.String JSON_CM_TYPE

See Also:

Constant Field Values

JSON_CM_HEADER

public static final java.lang.String JSON_CM_HEADER

See Also:

Constant Field Values

JSON_CM_MATRIX

public static final java.lang.String JSON_CM_MATRIX

See Also:

Constant Field Values

JSON_CM_TREES

public static final java.lang.String JSON_CM_TREES

See Also:

Constant Field Values

JSON_CM_CLASS_ERR

public static final java.lang.String JSON_CM_CLASS_ERR

See Also:

Constant Field Values

JSON_CM_ROWS

public static final java.lang.String JSON_CM_ROWS

See Also:

Constant Field Values

JSON_CM_ROWS_SKIPPED

public static final java.lang.String JSON_CM_ROWS_SKIPPED

See Also:

Constant Field Values

JSON_CM_CLASSES_ERRORS

public static final java.lang.String JSON_CM_CLASSES_ERRORS

See Also:

Constant Field Values

Constructor Detail

SpeeDRFModel

public SpeeDRFModel(Key selfKey,
            Key jobKey,
            Key dataKey,
            Frame fr,
            Vec response,
            Key[] t_keys,
            long zeed,
            java.lang.String[] cmDomain,
            SpeeDRF params)

Method Detail

get_params

public final SpeeDRF get_params()

Overrides:

get_params in class Model

job

public final Request2 job()

Overrides:

job in class Model

get_response

public Vec get_response()

treeCount

public int treeCount()

size

public int size()

classes

public int classes()

make

public static SpeeDRFModel make(SpeeDRFModel old,
                Key tkey,
                int nodeIdx)

name

public java.lang.String name(int atree)

tree

public byte[] tree(int tree_id)

Return the bits for a particular tree

delete_impl

public Futures delete_impl(Futures fs)

Free all internal tree keys.

Overrides:

delete_impl in class Model

classify0

public float classify0(int tree_id,
              Frame fr,
              Chunk[] chunks,
              int row,
              int[] modelDataMap,
              short badrow,
              boolean regression)

Classify a row according to one particular tree.

Parameters:

tree_id - the number of the tree to use

chunks - the chunk we are using

row - the row number in the chunk

modelDataMap - mapping from model/tree columns to data columns

Returns:

the predicted response class, or class+1 for broken rows

classify

public short classify(Frame fr,
             Chunk[] chks,
             int row,
             int[] modelDataMap,
             int[] votes,
             double[] classWt,
             java.util.Random rand)

classify

public short classify(int[] votes,
             double[] classWt,
             java.util.Random rand)

find_leaves_depth

public void find_leaves_depth()

Internal computation of depth and number of leaves.

leaves

public Counter leaves()

depth

public Counter depth()

colMap

public int[] colMap(java.lang.String[] names)

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

progress

public float progress()

Specified by:

progress in interface Job.Progress

generateHTML

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

generateHTMLTreeStats

public void generateHTMLTreeStats(java.lang.StringBuilder sb,
                         com.google.gson.JsonObject trees)

buildCM

public void buildCM(java.lang.StringBuilder sb)

makeAUC

protected static AUC makeAUC(ConfusionMatrix[] cms,
          float[] threshold,
          java.lang.String[] cmDomain)

generateHTMLAUC

protected void generateHTMLAUC(java.lang.StringBuilder sb)

generateHTMLVarImp

protected void generateHTMLVarImp(java.lang.StringBuilder sb)

doVarImpCalc

protected VarImp doVarImpCalc(SpeeDRFModel model)

computeVarImpSD

public static float[] computeVarImpSD(float[][] vote_diffs)