hex.genmodel.algos.xgboost

Class XGBoostNativeMojoModel

java.lang.Object

hex.genmodel.GenModel

hex.genmodel.MojoModel

hex.genmodel.algos.xgboost.XGBoostMojoModel

hex.genmodel.algos.xgboost.XGBoostNativeMojoModel

All Implemented Interfaces:

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

public final class XGBoostNativeMojoModel
extends XGBoostMojoModel

Please note: user is advised to explicitly release the native resources of XGBoost by calling close method on the instance.

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, _mojo_version, _nclasses, _nfeatures, _priorClassDistrib, _supervised, _uuid

Fields inherited from class hex.genmodel.GenModel

_domains, _names, _offsetColumn, _responseColumn

Constructor Summary

Constructors

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

Method Summary

Modifier and Type	Method and Description
void	close()
hex.genmodel.algos.tree.SharedTreeGraph	convert(int treeNumber, java.lang.String treeClass)
java.lang.String[]	getBoosterDump(boolean withStats, java.lang.String format)
static void	main(java.lang.String[] args)
double[]	score0(double[] doubles, double offset, double[] preds)
static double[]	score0(double[] doubles, double offset, double[] preds, ml.dmlc.xgboost4j.java.Booster _booster, int _nums, int _cats, int[] _catOffsets, boolean _useAllFactorLevels, int nclasses, double[] _priorClassDistrib, double _defaultThreshold, boolean _sparse)

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

_computeGraph, constructSubgraph, markOneHotEncodedCategoricals, postReadInit, 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, 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

XGBoostNativeMojoModel

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

Method Detail

score0

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

Overrides:

score0 in class hex.genmodel.GenModel

score0

public static double[] score0(double[] doubles,
                              double offset,
                              double[] preds,
                              ml.dmlc.xgboost4j.java.Booster _booster,
                              int _nums,
                              int _cats,
                              int[] _catOffsets,
                              boolean _useAllFactorLevels,
                              int nclasses,
                              double[] _priorClassDistrib,
                              double _defaultThreshold,
                              boolean _sparse)

close

public void close()

getBoosterDump

public java.lang.String[] getBoosterDump(boolean withStats,
                                         java.lang.String format)

main

public static void main(java.lang.String[] args)
                 throws java.io.IOException

Throws:

java.io.IOException

convert

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