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.SharedTreeGraphConverter, 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

_catOffsets, _cats, _featureMap, _nums, _sparse, _useAllFactorLevels

Fields inherited from class hex.genmodel.MojoModel

_balanceClasses, _category, _defaultThreshold, _h2oVersion, _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.PredictContributions	makeContributionsPredictor()
java.lang.Object	makeContributionsWorkspace()
void	postReadInit()
double[]	score0(double[] doubles, double offset, double[] preds)

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

_computeGraph, constructSubgraph, markOneHotEncodedCategoricals, 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, calibrateClassProbabilities, convertDouble2Float, correctProbabilities, createAuxKey, features, 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

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

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