org.apache.ignite.ml.knn.classification

Class KNNClassificationTrainer

java.lang.Object

org.apache.ignite.ml.knn.classification.KNNClassificationTrainer

All Implemented Interfaces:

DatasetTrainer<KNNClassificationModel,Double>, SingleLabelDatasetTrainer<KNNClassificationModel>

public class KNNClassificationTrainer
extends Object
implements SingleLabelDatasetTrainer<KNNClassificationModel>

kNN algorithm trainer to solve multi-class classification task.

Constructor Summary

Constructors

Constructor and Description
KNNClassificationTrainer()

Method Summary

Modifier and Type	Method and Description
<K,V> KNNClassificationModel	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

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

fit, fit

Constructor Detail

KNNClassificationTrainer

public KNNClassificationTrainer()

Method Detail

fit

public <K,V> KNNClassificationModel fit(DatasetBuilder<K,V> datasetBuilder,
                                        IgniteBiFunction<K,V,double[]> featureExtractor,
                                        IgniteBiFunction<K,V,Double> lbExtractor)

Trains model based on the specified data.

Specified by:

fit in interface DatasetTrainer<KNNClassificationModel,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.