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.IGenModel, java.io.Serializable, water.genmodel.IGeneratedModel

public final class XGBoostMojoModel
extends hex.genmodel.MojoModel

"Gradient Boosting Machine" MojoModel

See Also:

Serialized Form

Field Summary

Fields

Modifier and Type	Field and Description
int[]	_catOffsets
int	_cats
int	_nums
boolean	_sparse
boolean	_useAllFactorLevels

Fields inherited from class hex.genmodel.MojoModel

_balanceClasses, _category, _defaultThreshold, _modelClassDistrib, _nclasses, _nfeatures, _priorClassDistrib, _supervised, _uuid

Fields inherited from class hex.genmodel.GenModel

_domains, _names, _offsetColumn

Constructor Summary

Constructors

Constructor and Description
XGBoostMojoModel(java.lang.String[] columns, java.lang.String[][] domains)
XGBoostMojoModel(java.lang.String[] columns, java.lang.String[][] domains, ml.dmlc.xgboost4j.java.Booster _booster, int _nums, int _cats, int[] _catOffsets, boolean _useAllFactorLevels, boolean _sparse)

Method Summary

Methods

Modifier and Type	Method and Description
double[]	score0(double[] row, double[] preds)
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.MojoModel

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

Methods inherited from class hex.genmodel.GenModel

bitSetContains, bitSetIsInRange, calibrateClassProbabilities, 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_tweedieInv, img2pixels, isAutoEncoder, isClassifier, KMeans_closest, KMeans_distance, KMeans_distance, Kmeans_preprocessData, Kmeans_preprocessData, KMeans_simplex, log_rescale, mapEnum, setInput

Methods inherited from class java.lang.Object

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

Field Detail

_nums

public int _nums

_cats

public int _cats

_catOffsets

public int[] _catOffsets

_useAllFactorLevels

public boolean _useAllFactorLevels

_sparse

public boolean _sparse

Constructor Detail

XGBoostMojoModel

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

XGBoostMojoModel

public XGBoostMojoModel(java.lang.String[] columns,
                java.lang.String[][] domains,
                ml.dmlc.xgboost4j.java.Booster _booster,
                int _nums,
                int _cats,
                int[] _catOffsets,
                boolean _useAllFactorLevels,
                boolean _sparse)

Method Detail

score0

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

Specified by:

score0 in class hex.genmodel.GenModel

score0

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

Overrides:

score0 in class hex.genmodel.GenModel

score0

public static final 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)