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<M>

org.apache.ignite.ml.knn.KNNTrainer<KNNRegressionModel,KNNRegressionTrainer>

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

public class KNNRegressionTrainer
extends KNNTrainer<KNNRegressionModel,KNNRegressionTrainer>

KNN regression model trader that trains model on top of distribtued spatial indices. Be aware that this model is linked with cluster environment it's been built on and can't be saved or used in other places. Under the hood it keeps Dataset that consists of a set of resources allocated across the cluster.

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.knn.KNNTrainer

distanceMeasure, k, weighted

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

envBuilder, environment

Constructor Summary

Constructors

Constructor and Description
KNNRegressionTrainer()

Method Summary

Modifier and Type	Method and Description
protected KNNRegressionModel	convertDatasetIntoModel(Dataset<EmptyContext,SpatialIndex<Double>> dataset) Convers given dataset into KNN model (classification or regression depends on implementation).
protected KNNRegressionTrainer	self() Returns this instance.

Methods inherited from class org.apache.ignite.ml.knn.KNNTrainer

fitWithInitializedDeployingContext, isUpdateable, updateModel, withDataTtl, withDistanceMeasure, withIdxType, withK, withWeighted

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

fit, fit, fit, fit, fit, fit, getLastTrainedModelOrThrowEmptyDatasetException, identityTrainer, learningEnvironment, update, update, update, update, update, withConvertedLabels, withEnvironmentBuilder

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

convertDatasetIntoModel

protected KNNRegressionModel convertDatasetIntoModel(Dataset<EmptyContext,SpatialIndex<Double>> dataset)

Convers given dataset into KNN model (classification or regression depends on implementation).

Specified by:

convertDatasetIntoModel in class KNNTrainer<KNNRegressionModel,KNNRegressionTrainer>

Parameters:

dataset - Dataset of spatial indices.

Returns:

KNN model.

self

protected KNNRegressionTrainer self()

Returns this instance.

Specified by:

self in class KNNTrainer<KNNRegressionModel,KNNRegressionTrainer>

Returns:

This instance.