org.apache.ignite.ml.preprocessing.binarization

Class BinarizationPreprocessor<K,V>

java.lang.Object

org.apache.ignite.ml.preprocessing.binarization.BinarizationPreprocessor<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,Vector>, IgniteBiFunction<K,V,Vector>

public class BinarizationPreprocessor<K,V>
extends Object
implements IgniteBiFunction<K,V,Vector>

Preprocessing function that makes binarization. Feature values greater than the threshold are binarized to 1.0; values equal to or less than the threshold are binarized to 0.0.

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
BinarizationPreprocessor(double threshold, IgniteBiFunction<K,V,Vector> basePreprocessor) Constructs a new instance of Binarization preprocessor.

Method Summary

Modifier and Type	Method and Description
Vector	apply(K k, V v) Applies this preprocessor.
double	getThreshold() Get the threshold parameter.

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.math.functions.IgniteBiFunction

andThen

Methods inherited from interface java.util.function.BiFunction

andThen

Constructor Detail

BinarizationPreprocessor

public BinarizationPreprocessor(double threshold,
                                IgniteBiFunction<K,V,Vector> basePreprocessor)

Constructs a new instance of Binarization preprocessor.

Parameters:

threshold - Threshold value.

basePreprocessor - Base preprocessor.

Method Detail

apply

public Vector apply(K k,
                    V v)

Applies this preprocessor.

Specified by:

apply in interface BiFunction<K,V,Vector>

Parameters:

k - Key.

v - Value.

Returns:

Preprocessed row.

getThreshold

public double getThreshold()

Get the threshold parameter.