hex.genmodel.algos.tree

Class SharedTreeMojoModel

java.lang.Object

hex.genmodel.GenModel

hex.genmodel.MojoModel

hex.genmodel.algos.tree.SharedTreeMojoModel

All Implemented Interfaces:

IGenModel, java.io.Serializable, IGeneratedModel

Direct Known Subclasses:

DrfMojoModel, GbmMojoModel

public abstract class SharedTreeMojoModel
extends MojoModel

Common ancestor for DrfMojoModel and GbmMojoModel. See also: `hex.tree.SharedTreeModel` and `hex.tree.TreeVisitor` classes.

See Also:

Serialized Form

Field Summary

Fields

Modifier and Type	Field and Description
protected double[]	_calib_glm_beta GLM's beta used for calibrating output probabilities using Platt Scaling.
protected byte[][]	_compressed_trees Array of binary tree data, each tree being a byte[] array.
protected byte[][]	_compressed_trees_aux Array of auxiliary binary tree data, each being a byte[] array.
protected java.lang.Number	_mojo_version
protected int	_ntree_groups _ntree_groups is the number of trees requested by the user.
protected int	_ntrees_per_group

Fields inherited from class hex.genmodel.MojoModel

_balanceClasses, _category, _defaultThreshold, _modelClassDistrib, _nclasses, _nfeatures, _priorClassDistrib, _supervised, _uuid

Fields inherited from class hex.genmodel.GenModel

_domains, _names, _offsetColumn

Constructor Summary

Constructors

Modifier	Constructor and Description
protected	SharedTreeMojoModel(java.lang.String[] columns, java.lang.String[][] domains)

Method Summary

Methods

Modifier and Type	Method and Description
SharedTreeGraph	_computeGraph(int treeToPrint) Compute a graph of the forest.
boolean	calibrateClassProbabilities(double[] preds) Subclasses implement calibration of class probabilities.
static java.lang.String	getDecisionPath(double leafAssignment)
protected void	scoreAllTrees(double[] row, double[] preds) Score all trees and fill in the `preds` array.
static double	scoreTree(byte[] tree, double[] row, int nclasses, boolean computeLeafAssignment, java.lang.String[][] domains) Highly efficient (critical path) tree scoring Given a tree (in the form of a byte array) and the row of input data, compute either this tree's predicted value when `computeLeafAssignment` is false, or the the decision path within the tree (but no more than 64 levels) when `computeLeafAssignment` is true.
static double	scoreTree0(byte[] tree, double[] row, int nclasses, boolean computeLeafAssignment) SET IN STONE FOR MOJO VERSION "1.00" - DO NOT CHANGE
static double	scoreTree1(byte[] tree, double[] row, int nclasses, boolean computeLeafAssignment) SET IN STONE FOR MOJO VERSION "1.10" - DO NOT CHANGE
protected int	treeIndex(int groupIndex, int classIndex)

Methods inherited from class hex.genmodel.MojoModel

getModelCategory, getUUID, isSupervised, load, load, nclasses, nfeatures

Methods inherited from class hex.genmodel.GenModel

bitSetContains, bitSetIsInRange, correctProbabilities, createAuxKey, GBM_rescale, getColIdx, getDomainValues, getDomainValues, getDomainValues, getHeader, getModelCategories, getNames, getNumClasses, getNumCols, getNumResponseClasses, getPrediction, getPredsSize, getPredsSize, getResponseIdx, getResponseName, GLM_identityInv, GLM_inverseInv, GLM_logInv, GLM_logitInv, GLM_tweedieInv, img2pixels, isAutoEncoder, isClassifier, KMeans_closest, KMeans_distance, KMeans_distance, Kmeans_preprocessData, Kmeans_preprocessData, KMeans_simplex, log_rescale, mapEnum, score0, score0, setInput

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Field Detail

_mojo_version

protected java.lang.Number _mojo_version

_ntree_groups

protected int _ntree_groups

_ntree_groups is the number of trees requested by the user. For binomial case or regression this is also the total number of trees trained; however in multinomial case each requested "tree" is actually represented as a group of trees, with _ntrees_per_group trees in each group. Each of these individual trees assesses the likelihood that a given observation belongs to class A, B, C, etc. of a multiclass response.

_ntrees_per_group

protected int _ntrees_per_group

_compressed_trees

protected byte[][] _compressed_trees

Array of binary tree data, each tree being a byte[] array. The trees are logically grouped into a rectangular grid of dimensions _ntree_groups x _ntrees_per_group, however physically they are stored as 1-dimensional list, and an [i, j] logical tree is mapped to the index treeIndex(int, int).

_compressed_trees_aux

protected byte[][] _compressed_trees_aux

Array of auxiliary binary tree data, each being a byte[] array.

_calib_glm_beta

protected double[] _calib_glm_beta

GLM's beta used for calibrating output probabilities using Platt Scaling.

Constructor Detail

SharedTreeMojoModel

protected SharedTreeMojoModel(java.lang.String[] columns,
                   java.lang.String[][] domains)

Method Detail

scoreTree

public static double scoreTree(byte[] tree,
               double[] row,
               int nclasses,
               boolean computeLeafAssignment,
               java.lang.String[][] domains)

Highly efficient (critical path) tree scoring Given a tree (in the form of a byte array) and the row of input data, compute either this tree's predicted value when `computeLeafAssignment` is false, or the the decision path within the tree (but no more than 64 levels) when `computeLeafAssignment` is true. Note: this function is also used from the `hex.tree.CompressedTree` class in `h2o-algos` project.

getDecisionPath

public static java.lang.String getDecisionPath(double leafAssignment)

_computeGraph

public SharedTreeGraph _computeGraph(int treeToPrint)

Compute a graph of the forest.

Returns:

A graph of the forest.

scoreAllTrees

protected void scoreAllTrees(double[] row,
                 double[] preds)

Score all trees and fill in the `preds` array.

treeIndex

protected int treeIndex(int groupIndex,
            int classIndex)

scoreTree0

public static double scoreTree0(byte[] tree,
                double[] row,
                int nclasses,
                boolean computeLeafAssignment)

SET IN STONE FOR MOJO VERSION "1.00" - DO NOT CHANGE

Parameters:

tree -

row -

nclasses -

computeLeafAssignment -

Returns:

scoreTree1

public static double scoreTree1(byte[] tree,
                double[] row,
                int nclasses,
                boolean computeLeafAssignment)

SET IN STONE FOR MOJO VERSION "1.10" - DO NOT CHANGE

Parameters:

tree -

row -

nclasses -

computeLeafAssignment -

Returns:

calibrateClassProbabilities

public boolean calibrateClassProbabilities(double[] preds)

Description copied from class: GenModel

Subclasses implement calibration of class probabilities. The input is array of predictions returned by the scoring function (score0). Supports classification models that were trained with calibration enabled. Original probabilities in the predictions array are overwritten by their corresponding calibrated counterparts. Return false if model doesn't support calibration.

Overrides:

calibrateClassProbabilities in class GenModel