hex.genmodel.algos.targetencoder

Class TargetEncoderMojoModel

java.lang.Object

hex.genmodel.GenModel

hex.genmodel.MojoModel

hex.genmodel.algos.targetencoder.TargetEncoderMojoModel

All Implemented Interfaces:

IGenModel, java.io.Serializable, IGeneratedModel

public class TargetEncoderMojoModel
extends MojoModel

See Also:

Serialized Form

Field Summary

Fields

Modifier and Type	Field and Description
java.util.Map<java.lang.String,java.lang.Integer>	_columnNameToIdx
java.util.List<hex.genmodel.algos.targetencoder.ColumnsToSingleMapping>	_inencMapping
double	_inflectionPoint
java.util.List<ColumnsMapping>	_inoutMapping
double	_smoothing
java.util.Map<java.lang.String,java.lang.Boolean>	_teColumn2HasNAs
boolean	_withBlending

Fields inherited from class hex.genmodel.MojoModel

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

Fields inherited from class hex.genmodel.GenModel

_domains, _foldColumn, _names, _offsetColumn, _responseColumn, _treatmentColumn

Constructor Summary

Constructors

Constructor and Description
TargetEncoderMojoModel(java.lang.String[] columns, java.lang.String[][] domains, java.lang.String responseName)

Method Summary

Modifier and Type	Method and Description
static double	computeBlendedEncoding(double lambda, double posteriorMean, double priorMean)
static double	computeLambda(long nrows, double inflectionPoint, double smoothing)
EncodingMap	getEncodings(java.lang.String column)
int	getPredsSize() Returns the expected size of preds array which is passed to `predict(double[], double[])` function.
protected void	init()
double[]	score0(double[] row, double[] preds) Subclasses implement the scoring logic.
protected void	setEncodings(EncodingMaps encodingMaps)

Methods inherited from class hex.genmodel.MojoModel

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

Methods inherited from class hex.genmodel.GenModel

bitSetContains, bitSetIsInRange, calibrateClassProbabilities, convertDouble2Float, correctProbabilities, createAuxKey, features, GBM_rescale, getCategoricalEncoding, getColIdx, getDomainValues, getDomainValues, getDomainValues, getHeader, getModelCategories, getNames, getNumClasses, getNumCols, getNumResponseClasses, getOffsetName, getOrigDomainValues, getOrigNames, getOrigNumCols, getOrigProjectionArray, getOutputDomains, getOutputNames, getPrediction, getPredictionBinomial, getPredictionMultinomial, getPredsSize, getResponseIdx, getResponseName, GLM_identityInv, GLM_inverseInv, GLM_logInv, GLM_logitInv, GLM_ologitInv, GLM_tweedieInv, img2pixels, internal_threadSafeInstance, isAutoEncoder, isClassifier, KMeans_closest, KMeans_distance, KMeans_distance, KMeans_distances, Kmeans_preprocessData, Kmeans_preprocessData, KMeans_simplex, log_rescale, mapEnum, nCatFeatures, requiresOffset, score0, setCats, setCats, setInput, setInput

Methods inherited from class java.lang.Object

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

Field Detail

_columnNameToIdx

public final java.util.Map<java.lang.String,java.lang.Integer> _columnNameToIdx

_teColumn2HasNAs

public java.util.Map<java.lang.String,java.lang.Boolean> _teColumn2HasNAs

_withBlending

public boolean _withBlending

_inflectionPoint

public double _inflectionPoint

_smoothing

public double _smoothing

_inencMapping

public java.util.List<hex.genmodel.algos.targetencoder.ColumnsToSingleMapping> _inencMapping

_inoutMapping

public java.util.List<ColumnsMapping> _inoutMapping

Constructor Detail

TargetEncoderMojoModel

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

Method Detail

computeLambda

public static double computeLambda(long nrows,
                                   double inflectionPoint,
                                   double smoothing)

computeBlendedEncoding

public static double computeBlendedEncoding(double lambda,
                                            double posteriorMean,
                                            double priorMean)

init

protected void init()

setEncodings

protected void setEncodings(EncodingMaps encodingMaps)

getPredsSize

public int getPredsSize()

Description copied from class: GenModel

Returns the expected size of preds array which is passed to `predict(double[], double[])` function.

Specified by:

getPredsSize in interface IGeneratedModel

Overrides:

getPredsSize in class GenModel

Returns:

expected size of preds array

score0

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

Description copied from class: GenModel

Subclasses implement the scoring logic. The data is pre-loaded into a re-used temp array, in the order the model expects. The predictions are loaded into the re-used temp array, which is also returned. This call exactly matches the hex.Model.score0, but uses the light-weight GenModel class.

Specified by:

score0 in class GenModel

getEncodings

public EncodingMap getEncodings(java.lang.String column)