| Package | Description |
|---|---|
| org.apache.ignite.ml.knn |
Contains main APIs for kNN algorithms.
|
| org.apache.ignite.ml.knn.classification |
Contains main APIs for kNN classification algorithms.
|
| org.apache.ignite.ml.knn.regression |
Contains helper classes for kNN regression algorithms.
|
| org.apache.ignite.ml.structures |
Contains some internal utility structures.
|
| org.apache.ignite.ml.structures.partition |
Contains internal APIs for dataset partitioned labeled datasets.
|
| Modifier and Type | Method and Description |
|---|---|
static <K,V> Dataset<EmptyContext,LabeledDataset<Double,LabeledVector>> |
KNNUtils.buildDataset(DatasetBuilder<K,V> datasetBuilder,
IgniteBiFunction<K,V,double[]> featureExtractor,
IgniteBiFunction<K,V,Double> lbExtractor)
Builds dataset.
|
| Modifier and Type | Method and Description |
|---|---|
protected List<LabeledVector> |
KNNClassificationModel.findKNearestNeighbors(Vector v)
The main idea is calculation all distance pairs between given vector and all vectors in training set, sorting
them and finding k vectors with min distance with the given vector.
|
| Constructor and Description |
|---|
KNNClassificationModel(Dataset<EmptyContext,LabeledDataset<Double,LabeledVector>> dataset)
Builds the model via prepared dataset.
|
| Constructor and Description |
|---|
KNNRegressionModel(Dataset<EmptyContext,LabeledDataset<Double,LabeledVector>> dataset)
Builds the model via prepared dataset.
|
| Modifier and Type | Class and Description |
|---|---|
class |
LabeledDataset<L,Row extends LabeledVector>
Class for set of labeled vectors.
|
| Modifier and Type | Class and Description |
|---|---|
class |
LabeledVectorDouble<V extends Vector>
Labeled vector specialized to double label.
|
| Constructor and Description |
|---|
LabeledDataset(Row[] data)
Creates new Labeled Dataset by given data.
|
LabeledDataset(Row[] data,
int colSize)
Creates new Labeled Dataset by given data.
|
| Modifier and Type | Method and Description |
|---|---|
LabeledDataset<Double,LabeledVector> |
LabeledDatasetPartitionDataBuilderOnHeap.build(Iterator<UpstreamEntry<K,V>> upstreamData,
long upstreamDataSize,
C ctx)
Builds a new partition
data from a partition upstream data and partition context |
Follow @ApacheIgnite
Ignite Fabric : ver. 2.6.0 Release Date : July 10 2018