hex.genmodel.algos.xgboost

Class XGBoostMojoModel

java.lang.Object

hex.genmodel.GenModel

hex.genmodel.MojoModel

hex.genmodel.algos.xgboost.XGBoostMojoModel

All Implemented Interfaces:

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

Direct Known Subclasses:

XGBoostJavaMojoModel, XGBoostNativeMojoModel

public abstract class XGBoostMojoModel
extends hex.genmodel.MojoModel
implements hex.genmodel.algos.tree.SharedTreeGraphConverter, java.io.Closeable

"Gradient Boosting Machine" MojoModel

See Also:

Serialized Form

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	XGBoostMojoModel.ObjectiveType

Field Summary

Fields

Modifier and Type	Field and Description
int[]	_catOffsets
int	_cats
java.lang.String	_featureMap
int	_nums
boolean	_sparse
boolean	_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
XGBoostMojoModel(java.lang.String[] columns, java.lang.String[][] domains, java.lang.String responseColumn)

Method Summary

Modifier and Type	Method and Description
protected hex.genmodel.algos.tree.SharedTreeGraph	_computeGraph(biz.k11i.xgboost.gbm.GradBooster booster, int treeNumber)
protected void	constructSubgraph(biz.k11i.xgboost.tree.RegTreeNode[] xgBoostNodes, hex.genmodel.algos.tree.SharedTreeNode sharedTreeNode, int nodeIndex, hex.genmodel.algos.tree.SharedTreeSubgraph sharedTreeSubgraph, boolean[] oneHotEncodedMap, boolean inclusiveNA, java.lang.String[] features)
protected boolean[]	markOneHotEncodedCategoricals(java.lang.String[] featureMap)
void	postReadInit()
double[]	score0(double[] row, double[] preds)
static double[]	toPreds(double[] in, float[] out, double[] preds, int nclasses, double[] priorClassDistrib, double defaultThreshold)

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, score0, setCats, setCats, setInput, setInput

Methods inherited from class java.lang.Object

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

Methods inherited from interface hex.genmodel.algos.tree.SharedTreeGraphConverter

convert

Methods inherited from interface java.io.Closeable

close

Field Detail

_nums

public int _nums

_cats

public int _cats

_catOffsets

public int[] _catOffsets

_useAllFactorLevels

public boolean _useAllFactorLevels

_sparse

public boolean _sparse

_featureMap

public java.lang.String _featureMap

Constructor Detail

XGBoostMojoModel

public XGBoostMojoModel(java.lang.String[] columns,
                        java.lang.String[][] domains,
                        java.lang.String responseColumn)

Method Detail

postReadInit

public void postReadInit()

score0

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

Specified by:

score0 in class hex.genmodel.GenModel

toPreds

public static double[] toPreds(double[] in,
                               float[] out,
                               double[] preds,
                               int nclasses,
                               double[] priorClassDistrib,
                               double defaultThreshold)

constructSubgraph

protected void constructSubgraph(biz.k11i.xgboost.tree.RegTreeNode[] xgBoostNodes,
                                 hex.genmodel.algos.tree.SharedTreeNode sharedTreeNode,
                                 int nodeIndex,
                                 hex.genmodel.algos.tree.SharedTreeSubgraph sharedTreeSubgraph,
                                 boolean[] oneHotEncodedMap,
                                 boolean inclusiveNA,
                                 java.lang.String[] features)

markOneHotEncodedCategoricals

protected boolean[] markOneHotEncodedCategoricals(java.lang.String[] featureMap)

_computeGraph

protected hex.genmodel.algos.tree.SharedTreeGraph _computeGraph(biz.k11i.xgboost.gbm.GradBooster booster,
                                                                int treeNumber)