org.apache.ignite.ml.preprocessing.imputing

Class ImputerTrainer<K,V>

java.lang.Object

org.apache.ignite.ml.preprocessing.imputing.ImputerTrainer<K,V>

Type Parameters:

K - Type of a key in upstream data.

V - Type of a value in upstream data.

All Implemented Interfaces:

PreprocessingTrainer<K,V,Vector,Vector>

public class ImputerTrainer<K,V>
extends Object
implements PreprocessingTrainer<K,V,Vector,Vector>

Trainer of the imputing preprocessor. The imputing fills the missed values according the imputing strategy (default: mean value for each feature). It supports double values in features only.

Constructor Summary

Constructors

Constructor and Description
ImputerTrainer()

Method Summary

Modifier and Type	Method and Description
ImputerPreprocessor<K,V>	fit(DatasetBuilder<K,V> datasetBuilder, IgniteBiFunction<K,V,Vector> basePreprocessor) Fits preprocessor.
ImputerTrainer<K,V>	withImputingStrategy(ImputingStrategy imputingStgy) Sets the imputing strategy.

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.PreprocessingTrainer

fit, fit

Constructor Detail

ImputerTrainer

public ImputerTrainer()

Method Detail

fit

public ImputerPreprocessor<K,V> fit(DatasetBuilder<K,V> datasetBuilder,
                                    IgniteBiFunction<K,V,Vector> basePreprocessor)

Fits preprocessor.

Specified by:

fit in interface PreprocessingTrainer<K,V,Vector,Vector>

Parameters:

datasetBuilder - Dataset builder.

basePreprocessor - Base preprocessor.

Returns:

Preprocessor.

withImputingStrategy

public ImputerTrainer<K,V> withImputingStrategy(ImputingStrategy imputingStgy)

Sets the imputing strategy.

Parameters:

imputingStgy - The given value.

Returns:

The updated imputing trainer.