public final class GbmMojoModel extends SharedTreeMojoModel implements SharedTreeGraphConverter
SharedTreeMojoModel.DecisionPathTracker<T>, SharedTreeMojoModel.LeafDecisionPathTracker, SharedTreeMojoModel.LeafNodeAssignments, SharedTreeMojoModel.StringDecisionPathTracker
Modifier and Type | Field and Description |
---|---|
DistributionFamily |
_family |
double |
_init_f |
_calib_glm_beta, _compressed_trees, _compressed_trees_aux, _ntree_groups, _ntrees_per_group
_balanceClasses, _category, _defaultThreshold, _h2oVersion, _modelClassDistrib, _mojo_version, _nclasses, _nfeatures, _priorClassDistrib, _supervised, _uuid
_domains, _names, _offsetColumn, _responseColumn
Constructor and Description |
---|
GbmMojoModel(java.lang.String[] columns,
java.lang.String[][] domains,
java.lang.String responseColumn) |
Modifier and Type | Method and Description |
---|---|
java.lang.String[] |
leaf_node_assignment(double[] row) |
double[] |
score0(double[] row,
double[] preds)
Subclasses implement the scoring logic.
|
double[] |
score0(double[] row,
double offset,
double[] preds)
Corresponds to `hex.tree.gbm.GbmMojoModel.score0()`
|
double[] |
unifyPreds(double[] row,
double offset,
double[] preds)
Transforms tree predictions into the final model predictions.
|
_computeGraph, calibrateClassProbabilities, computeTreeGraph, convert, getDecisionPath, getDecisionPath, getDecisionPath, getDecisionPathNames, getLeafNodeAssignments, getLeafNodeId, getNTreeGroups, getNTreesPerGroup, postInit, scoreAllTrees, scoreSingleTree, scoreStagedPredictions, scoreTree, scoreTree, scoreTree0, scoreTree0, scoreTree1, scoreTree1, scoreTreeRange, treeName
getModelCategory, getUUID, isSupervised, load, load, nclasses, nfeatures
bitSetContains, bitSetIsInRange, convertDouble2Float, 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_ologitInv, GLM_tweedieInv, img2pixels, isAutoEncoder, isClassifier, KMeans_closest, KMeans_distance, KMeans_distance, KMeans_distances, Kmeans_preprocessData, Kmeans_preprocessData, KMeans_simplex, log_rescale, mapEnum, setCats, setCats, setInput, setInput
clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait
convert
public DistributionFamily _family
public double _init_f
public GbmMojoModel(java.lang.String[] columns, java.lang.String[][] domains, java.lang.String responseColumn)
public final double[] score0(double[] row, double offset, double[] preds)
public final double[] unifyPreds(double[] row, double offset, double[] preds)
SharedTreeMojoModel
unifyPreds
in class SharedTreeMojoModel
row
- input row.offset
- offset.preds
- final output, same structure as of GenModel.score0(double[], double[])
.public double[] score0(double[] row, double[] preds)
GenModel
public java.lang.String[] leaf_node_assignment(double[] row)