org.apache.ignite.ml.knn

Class KNNUtils

java.lang.Object

org.apache.ignite.ml.knn.KNNUtils

public class KNNUtils
extends Object

Helper class for KNNRegression.

Constructor Summary

Constructors

Constructor and Description
KNNUtils()

Method Summary

Modifier and Type	Method and Description
static <K,V> @Nullable Dataset<EmptyContext,LabeledVectorSet<Double,LabeledVector>>	buildDataset(DatasetBuilder<K,V> datasetBuilder, IgniteBiFunction<K,V,Vector> featureExtractor, IgniteBiFunction<K,V,Double> lbExtractor) Builds dataset.

Methods inherited from class java.lang.Object

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

Constructor Detail

KNNUtils

public KNNUtils()

Method Detail

buildDataset

@Nullable
public static <K,V> @Nullable Dataset<EmptyContext,LabeledVectorSet<Double,LabeledVector>> buildDataset(DatasetBuilder<K,V> datasetBuilder,
                                                                                                                  IgniteBiFunction<K,V,Vector> featureExtractor,
                                                                                                                  IgniteBiFunction<K,V,Double> lbExtractor)

Builds dataset.

Parameters:

datasetBuilder - Dataset builder.

featureExtractor - Feature extractor.

lbExtractor - Label extractor.

Returns:

Dataset.