hex.genmodel.algos.xgboost

Class XGBoostJavaMojoModel

java.lang.Object

hex.genmodel.GenModel

hex.genmodel.MojoModel

hex.genmodel.algos.xgboost.XGBoostMojoModel

hex.genmodel.algos.xgboost.XGBoostJavaMojoModel

All Implemented Interfaces:

hex.genmodel.algos.tree.PlattScalingMojoHelper.MojoModelWithCalibration, hex.genmodel.algos.tree.SharedTreeGraphConverter, hex.genmodel.algos.tree.TreeBackedMojoModel, hex.genmodel.IGenModel, hex.genmodel.PredictContributionsFactory, java.io.Closeable, java.io.Serializable, java.lang.AutoCloseable, water.genmodel.IGeneratedModel

public final class XGBoostJavaMojoModel
extends XGBoostMojoModel
implements hex.genmodel.PredictContributionsFactory

Implementation of XGBoostMojoModel that uses Pure Java Predict see https://github.com/h2oai/xgboost-predictor

See Also:

Serialized Form

Nested Class Summary

Nested classes/interfaces inherited from class hex.genmodel.algos.xgboost.XGBoostMojoModel

XGBoostMojoModel.ObjectiveType

Field Summary

Fields inherited from class hex.genmodel.algos.xgboost.XGBoostMojoModel

_boosterType, _calib_glm_beta, _catOffsets, _cats, _featureMap, _hasOffset, _ntrees, _nums, _sparse, _useAllFactorLevels

Fields inherited from class hex.genmodel.MojoModel

_algoName, _balanceClasses, _category, _defaultThreshold, _h2oVersion, _modelAttributes, _modelClassDistrib, _modelDescriptor, _mojo_version, _nclasses, _nfeatures, _priorClassDistrib, _supervised, _uuid

Fields inherited from class hex.genmodel.GenModel

_domains, _names, _offsetColumn, _responseColumn

Constructor Summary

Constructors

Constructor and Description
XGBoostJavaMojoModel(byte[] boosterBytes, java.lang.String[] columns, java.lang.String[][] domains, java.lang.String responseColumn)
XGBoostJavaMojoModel(byte[] boosterBytes, java.lang.String[] columns, java.lang.String[][] domains, java.lang.String responseColumn, boolean enableTreeSHAP)

Method Summary

Modifier and Type	Method and Description
float[]	calculateContributions(biz.k11i.xgboost.util.FVec row, float[] out_contribs, java.lang.Object workspace)
void	close()
hex.genmodel.algos.tree.SharedTreeGraph	convert(int treeNumber, java.lang.String treeClass)
hex.genmodel.algos.tree.SharedTreeGraph	convert(int treeNumber, java.lang.String treeClass, hex.genmodel.algos.tree.ConvertTreeOptions options)
java.lang.String[]	getDecisionPath(double[] doubles)
double	getInitF()
hex.genmodel.algos.tree.SharedTreeMojoModel.LeafNodeAssignments	getLeafNodeAssignments(double[] doubles)
hex.genmodel.PredictContributions	makeContributionsPredictor()
java.lang.Object	makeContributionsWorkspace()
static biz.k11i.xgboost.Predictor	makePredictor(byte[] boosterBytes)
void	postReadInit()
double[]	score0(double[] doubles, double offset, double[] preds)

Methods inherited from class hex.genmodel.algos.xgboost.XGBoostMojoModel

calibrateClassProbabilities, constructSubgraph, getCalibGlmBeta, getNTreeGroups, getNTreesPerGroup, markOneHotEncodedCategoricals, requiresOffset, score0, toPreds

Methods inherited from class hex.genmodel.MojoModel

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

Methods inherited from class hex.genmodel.GenModel

bitSetContains, bitSetIsInRange, convertDouble2Float, correctProbabilities, createAuxKey, features, GBM_rescale, getCategoricalEncoding, getColIdx, getDomainValues, getDomainValues, getDomainValues, getHeader, getModelCategories, getNames, getNumClasses, getNumCols, getNumResponseClasses, getOffsetName, getOrigDomainValues, getOrigNames, getOrigNumCols, getOrigProjectionArray, 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, nCatFeatures, setCats, setCats, setInput, setInput

Methods inherited from class java.lang.Object

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

Constructor Detail

XGBoostJavaMojoModel

public XGBoostJavaMojoModel(byte[] boosterBytes,
                            java.lang.String[] columns,
                            java.lang.String[][] domains,
                            java.lang.String responseColumn)

XGBoostJavaMojoModel

public XGBoostJavaMojoModel(byte[] boosterBytes,
                            java.lang.String[] columns,
                            java.lang.String[][] domains,
                            java.lang.String responseColumn,
                            boolean enableTreeSHAP)

Method Detail

postReadInit

public void postReadInit()

Overrides:

postReadInit in class XGBoostMojoModel

makePredictor

public static biz.k11i.xgboost.Predictor makePredictor(byte[] boosterBytes)

score0

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

Overrides:

score0 in class hex.genmodel.GenModel

makeContributionsWorkspace

public final java.lang.Object makeContributionsWorkspace()

calculateContributions

public final float[] calculateContributions(biz.k11i.xgboost.util.FVec row,
                                            float[] out_contribs,
                                            java.lang.Object workspace)

makeContributionsPredictor

public final hex.genmodel.PredictContributions makeContributionsPredictor()

Specified by:

makeContributionsPredictor in interface hex.genmodel.PredictContributionsFactory

close

public void close()

Specified by:

close in interface java.io.Closeable

Specified by:

close in interface java.lang.AutoCloseable

convert

public hex.genmodel.algos.tree.SharedTreeGraph convert(int treeNumber,
                                                       java.lang.String treeClass)

Specified by:

convert in interface hex.genmodel.algos.tree.SharedTreeGraphConverter

convert

public hex.genmodel.algos.tree.SharedTreeGraph convert(int treeNumber,
                                                       java.lang.String treeClass,
                                                       hex.genmodel.algos.tree.ConvertTreeOptions options)

Specified by:

convert in interface hex.genmodel.algos.tree.SharedTreeGraphConverter

Overrides:

convert in class XGBoostMojoModel

getInitF

public double getInitF()

Specified by:

getInitF in interface hex.genmodel.algos.tree.TreeBackedMojoModel

getLeafNodeAssignments

public hex.genmodel.algos.tree.SharedTreeMojoModel.LeafNodeAssignments getLeafNodeAssignments(double[] doubles)

Specified by:

getLeafNodeAssignments in interface hex.genmodel.algos.tree.TreeBackedMojoModel

getDecisionPath

public java.lang.String[] getDecisionPath(double[] doubles)

Specified by:

getDecisionPath in interface hex.genmodel.algos.tree.TreeBackedMojoModel