org.deeplearning4j.clustering.kdtree

Class KDTree

java.lang.Object

org.deeplearning4j.clustering.kdtree.KDTree

All Implemented Interfaces:

java.io.Serializable

public class KDTree
extends java.lang.Object
implements java.io.Serializable

KDTree based on: https://github.com/nicky-zs/kdtree-python/blob/master/kdtree.py

See Also:

Serialized Form

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	KDTree.KDNode

Field Summary

Fields

Modifier and Type	Field and Description
static int	GREATER
static int	LESS

Constructor Summary

Constructors

Constructor and Description
KDTree(int dims)

Method Summary

Modifier and Type	Method and Description
KDTree.KDNode	delete(org.nd4j.linalg.api.ndarray.INDArray point)
void	insert(org.nd4j.linalg.api.ndarray.INDArray point) Insert a point in to the tree
java.util.List<Pair<java.lang.Double,org.nd4j.linalg.api.ndarray.INDArray>>	knn(org.nd4j.linalg.api.ndarray.INDArray point, double distance)
Pair<java.lang.Double,org.nd4j.linalg.api.ndarray.INDArray>	nn(org.nd4j.linalg.api.ndarray.INDArray point) Query for nearest neighbor.
int	size() The number of elements in the tree

Methods inherited from class java.lang.Object

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

Field Detail

GREATER

public static final int GREATER

See Also:

Constant Field Values

LESS

public static final int LESS

See Also:

Constant Field Values

Constructor Detail

KDTree

public KDTree(int dims)

Method Detail

insert

public void insert(org.nd4j.linalg.api.ndarray.INDArray point)

Insert a point in to the tree

Parameters:

point - the point to insert

delete

public KDTree.KDNode delete(org.nd4j.linalg.api.ndarray.INDArray point)

knn

public java.util.List<Pair<java.lang.Double,org.nd4j.linalg.api.ndarray.INDArray>> knn(org.nd4j.linalg.api.ndarray.INDArray point,
                                                                                       double distance)

nn

public Pair<java.lang.Double,org.nd4j.linalg.api.ndarray.INDArray> nn(org.nd4j.linalg.api.ndarray.INDArray point)

Query for nearest neighbor. Returns the distance and point

Parameters:

point - the point to query for

Returns:

size

public int size()

The number of elements in the tree

Returns:

the number of elements in the tree