org.apache.ignite.ml.knn.regression

Class KNNRegressionTrainer

java.lang.Object

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

public class KNNRegressionTrainer
extends Object

kNN algorithm trainer to solve regression task.

Constructor Summary

Constructors

Constructor and Description
KNNRegressionTrainer()

Method Summary

Modifier and Type	Method and Description
<K,V> KNNRegressionModel	fit(DatasetBuilder<K,V> datasetBuilder, IgniteBiFunction<K,V,double[]> featureExtractor, IgniteBiFunction<K,V,Double> lbExtractor) Trains model based on the specified data.

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,double[]> featureExtractor,
                                    IgniteBiFunction<K,V,Double> lbExtractor)

Trains model based on the specified data.

Parameters:

datasetBuilder - Dataset builder.

featureExtractor - Feature extractor.

lbExtractor - Label extractor.

Returns:

Model.