public class KDTreeFloatEuclideanAssigner extends Object implements HardAssigner<float[],float[],IntFloatPair>
HardAssigner
that uses a FloatNearestNeighboursKDTree
to
generate approximately correct cluster assignments.Modifier and Type | Field and Description |
---|---|
protected FloatNearestNeighboursKDTree |
nn |
Constructor and Description |
---|
KDTreeFloatEuclideanAssigner(CentroidsProvider<float[]> provider)
Construct the assigner using the given cluster data.
|
KDTreeFloatEuclideanAssigner(float[][] data)
Construct the assigner using the given cluster data.
|
Modifier and Type | Method and Description |
---|---|
int |
assign(float[] data)
Assign a single point to a cluster.
|
int[] |
assign(float[][] data)
Assign data to a cluster.
|
IntFloatPair |
assignDistance(float[] data)
Assign a single point to a cluster.
|
void |
assignDistance(float[][] data,
int[] indices,
float[] distances)
Assign data to clusters.
|
int |
numDimensions()
Get the number of dimensions of the input vectors.
|
int |
size()
The number of centroids or unique ids that can be generated.
|
protected FloatNearestNeighboursKDTree nn
public KDTreeFloatEuclideanAssigner(CentroidsProvider<float[]> provider)
provider
- the cluster data providerpublic KDTreeFloatEuclideanAssigner(float[][] data)
data
- the cluster datapublic int[] assign(float[][] data)
HardAssigner
assign
in interface HardAssigner<float[],float[],IntFloatPair>
data
- the data.public int assign(float[] data)
HardAssigner
assign
in interface HardAssigner<float[],float[],IntFloatPair>
data
- datum to assign.public void assignDistance(float[][] data, int[] indices, float[] distances)
HardAssigner
assignDistance
in interface HardAssigner<float[],float[],IntFloatPair>
data
- the data.indices
- the cluster index for each data point.distances
- the distance to the closest cluster for each data point.public IntFloatPair assignDistance(float[] data)
HardAssigner
assignDistance
in interface HardAssigner<float[],float[],IntFloatPair>
data
- point to assign.public int size()
HardAssigner
size
in interface HardAssigner<float[],float[],IntFloatPair>
public int numDimensions()
Assigner
numDimensions
in interface Assigner<float[]>