org.apache.ignite.ml.knn.regression

Class KNNRegressionTrainer

java.lang.Object

org.apache.ignite.ml.trainers.DatasetTrainer<M,Double>

org.apache.ignite.ml.trainers.SingleLabelDatasetTrainer<KNNRegressionModel>

org.apache.ignite.ml.knn.regression.KNNRegressionTrainer

public class KNNRegressionTrainer
extends SingleLabelDatasetTrainer<KNNRegressionModel>

kNN algorithm trainer to solve regression task.

Nested Class Summary

Nested classes/interfaces inherited from class org.apache.ignite.ml.trainers.DatasetTrainer

DatasetTrainer.EmptyDatasetException

Field Summary

Fields inherited from class org.apache.ignite.ml.trainers.DatasetTrainer

environment

Constructor Summary

Constructors

Constructor and Description
KNNRegressionTrainer()

Method Summary

Modifier and Type	Method and Description
protected boolean	checkState(KNNRegressionModel mdl)
<K,V> KNNRegressionModel	fit(DatasetBuilder<K,V> datasetBuilder, IgniteBiFunction<K,V,Vector> featureExtractor, IgniteBiFunction<K,V,Double> lbExtractor) Trains model based on the specified data.
<K,V> KNNRegressionModel	updateModel(KNNRegressionModel mdl, DatasetBuilder<K,V> datasetBuilder, IgniteBiFunction<K,V,Vector> featureExtractor, IgniteBiFunction<K,V,Double> lbExtractor) Gets state of model in arguments, update in according to new data and return new model.

Methods inherited from class org.apache.ignite.ml.trainers.DatasetTrainer

fit, fit, fit, fit, getLastTrainedModelOrThrowEmptyDatasetException, setEnvironment, update, update, update, update, update

Methods inherited from class java.lang.Object

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

Constructor Detail

KNNRegressionTrainer

public KNNRegressionTrainer()

Method Detail

fit

public <K,V> KNNRegressionModel fit(DatasetBuilder<K,V> datasetBuilder,
                                    IgniteBiFunction<K,V,Vector> featureExtractor,
                                    IgniteBiFunction<K,V,Double> lbExtractor)

Trains model based on the specified data.

Specified by:

fit in class DatasetTrainer<KNNRegressionModel,Double>

Type Parameters:

K - Type of a key in upstream data.

V - Type of a value in upstream data.

Parameters:

datasetBuilder - Dataset builder.

featureExtractor - Feature extractor.

lbExtractor - Label extractor.

Returns:

Model.

updateModel

public <K,V> KNNRegressionModel updateModel(KNNRegressionModel mdl,
                                            DatasetBuilder<K,V> datasetBuilder,
                                            IgniteBiFunction<K,V,Vector> featureExtractor,
                                            IgniteBiFunction<K,V,Double> lbExtractor)

Gets state of model in arguments, update in according to new data and return new model.

Specified by:

updateModel in class DatasetTrainer<KNNRegressionModel,Double>

Type Parameters:

K - Type of a key in upstream data.

V - Type of a value in upstream data.

Parameters:

mdl - Learned model.

datasetBuilder - Dataset builder.

featureExtractor - Feature extractor.

lbExtractor - Label extractor.

Returns:

Updated model.

checkState

protected boolean checkState(KNNRegressionModel mdl)

Specified by:

checkState in class DatasetTrainer<KNNRegressionModel,Double>

Parameters:

mdl - Model.

Returns:

true if current critical for training parameters correspond to parameters from last training.