hex.genmodel.algos

Class GbmModel

java.lang.Object

hex.genmodel.GenModel

hex.genmodel.MojoModel

hex.genmodel.algos.TreeBasedModel

hex.genmodel.algos.GbmModel

All Implemented Interfaces:

IGenModel, java.io.Serializable, IGeneratedModel

public final class GbmModel
extends TreeBasedModel

"Gradient Boosting Machine" MojoModel

See Also:

Serialized Form

Nested Class Summary

Nested classes/interfaces inherited from class hex.genmodel.MojoModel

MojoModel.MojoReader

Field Summary

Fields inherited from class hex.genmodel.algos.TreeBasedModel

_compressed_trees, _ntrees

Fields inherited from class hex.genmodel.MojoModel

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

Fields inherited from class hex.genmodel.GenModel

_domains, _names, _offsetColumn

Constructor Summary

Constructors

Constructor and Description
GbmModel(MojoModel.MojoReader cr, java.util.Map<java.lang.String,java.lang.Object> info, java.lang.String[] columns, java.lang.String[][] domains)

Method Summary

Methods

Modifier and Type	Method and Description
double[]	score0(double[] row, double[] preds) Subclasses implement the scoring logic.
double[]	score0(double[] row, double offset, double[] preds) Corresponds to `hex.tree.drf.DrfModel.score0()`

Methods inherited from class hex.genmodel.algos.TreeBasedModel

getDecisionPath, scoreAllTrees, scoreTree, scoreTree

Methods inherited from class hex.genmodel.MojoModel

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

Methods inherited from class hex.genmodel.GenModel

bitSetContains, correctProbabilities, GBM_rescale, getColIdx, getDomainValues, getDomainValues, getDomainValues, getHeader, getNames, getNumClasses, getNumCols, getNumResponseClasses, getPrediction, getPredsSize, getResponseIdx, getResponseName, GLM_identityInv, GLM_inverseInv, GLM_logInv, GLM_logitInv, GLM_tweedieInv, isAutoEncoder, isClassifier, KMeans_closest, KMeans_distance, KMeans_distance, KMeans_simplex, log_rescale, map, mapEnum, predict, predict, score0, score0, score0

Methods inherited from class java.lang.Object

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

Constructor Detail

GbmModel

public GbmModel(MojoModel.MojoReader cr,
        java.util.Map<java.lang.String,java.lang.Object> info,
        java.lang.String[] columns,
        java.lang.String[][] domains)

Method Detail

score0

public final double[] score0(double[] row,
              double offset,
              double[] preds)

Corresponds to `hex.tree.drf.DrfModel.score0()`

Overrides:

score0 in class GenModel

score0

public double[] score0(double[] row,
              double[] preds)

Description copied from class: GenModel

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. This call exactly matches the hex.Model.score0, but uses the light-weight GenModel class.

Specified by:

score0 in class GenModel