hex.genmodel.easy

Class EnumEncoder

java.lang.Object

hex.genmodel.easy.EnumEncoder

All Implemented Interfaces:

CategoricalEncoder, java.io.Serializable

public class EnumEncoder
extends java.lang.Object
implements CategoricalEncoder

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
EnumEncoder(java.lang.String columnName, int targetIndex, java.lang.String[] domainValues)

Method Summary

Modifier and Type	Method and Description
boolean	encodeCatValue(java.lang.String levelName, double[] rawData) Encodes a given categorical level into a raw array onto the right position.
void	encodeNA(double[] rawData) Encode NA (missing level) into raw data.

Methods inherited from class java.lang.Object

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

Constructor Detail

EnumEncoder

public EnumEncoder(java.lang.String columnName,
                   int targetIndex,
                   java.lang.String[] domainValues)

Method Detail

encodeCatValue

public boolean encodeCatValue(java.lang.String levelName,
                              double[] rawData)

Description copied from interface: CategoricalEncoder

Encodes a given categorical level into a raw array onto the right position.

Specified by:

encodeCatValue in interface CategoricalEncoder

Parameters:

levelName - categorical level

rawData - raw input to score0

Returns:

true if provided categorical level was valid and was properly encoded, false if nothing was written to raw data

encodeNA

public void encodeNA(double[] rawData)

Description copied from interface: CategoricalEncoder

Encode NA (missing level) into raw data.

Specified by:

encodeNA in interface CategoricalEncoder

Parameters:

rawData - target raw data array