org.apache.ignite.ml.preprocessing.encoding.label

Class LabelEncoderPreprocessor<K,V>

java.lang.Object

org.apache.ignite.ml.preprocessing.encoding.EncoderPreprocessor<K,V>

org.apache.ignite.ml.preprocessing.encoding.label.LabelEncoderPreprocessor<K,V>

Type Parameters:

K - Type of a key in upstream data.

V - Type of a value in upstream data.

All Implemented Interfaces:

Serializable, BiFunction<K,V,LabeledVector>, DeployableObject, IgniteBiFunction<K,V,LabeledVector>, Preprocessor<K,V>

public final class LabelEncoderPreprocessor<K,V>
extends EncoderPreprocessor<K,V>
implements DeployableObject

Preprocessing function that makes Label encoding. The Label Encoder Preprocessor encodes string values (categories) to double values in range [0.0, amountOfCategories), where the most popular value will be presented as 0.0 and the least popular value presented with amountOfCategories-1 value.

This preprocessor can transforms label column.

NOTE: it does not add new column but change data in-place.

See Also:

Serialized Form

Field Summary

Fields inherited from class org.apache.ignite.ml.preprocessing.encoding.EncoderPreprocessor

basePreprocessor, encodingValues, handledIndices, KEY_FOR_NULL_VALUES, labelFrequencies

Constructor Summary

Constructors

Constructor and Description
LabelEncoderPreprocessor(Map<String,Integer> labelFrequencies, Preprocessor<K,V> basePreprocessor) Constructs a new instance of Label Encoder preprocessor.

Method Summary

Modifier and Type	Method and Description
LabeledVector	apply(K k, V v) Applies this preprocessor.
List<Object>	getDependencies() Returns dependencies of this object that can be object with class defined by client side and unknown for server.

Methods inherited from class java.lang.Object

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

Methods inherited from interface org.apache.ignite.ml.preprocessing.Preprocessor

map

Methods inherited from interface org.apache.ignite.ml.math.functions.IgniteBiFunction

andThen

Methods inherited from interface java.util.function.BiFunction

andThen

Constructor Detail

LabelEncoderPreprocessor

public LabelEncoderPreprocessor(Map<String,Integer> labelFrequencies,
                                Preprocessor<K,V> basePreprocessor)

Constructs a new instance of Label Encoder preprocessor.

Parameters:

basePreprocessor - Base preprocessor.

Method Detail

apply

public LabeledVector apply(K k,
                           V v)

Applies this preprocessor.

Specified by:

apply in interface BiFunction<K,V,LabeledVector>

Parameters:

k - Key.

v - Value.

Returns:

Preprocessed row.

getDependencies

public List<Object> getDependencies()

Returns dependencies of this object that can be object with class defined by client side and unknown for server.

Specified by:

getDependencies in interface DeployableObject