Constructor Detail

• RPForest

  public RPForest(int numTrees,
                  int maxSize,
                  String similarityFunction)

  Create the rp forest with the specified number of trees

  Parameters:

  numTrees - the number of trees in the forest

  maxSize - the max size of each tree

  similarityFunction - the distance function to use

Method Detail

• fit

  public void fit(INDArray x)

  Build the trees from the given dataset

  Parameters:

  x - the input dataset (should be a 2d matrix)

• getAllCandidates

  public INDArray getAllCandidates(INDArray input)

  Get all candidates relative to a specific datapoint.

  Parameters:

  input -

  Returns:

• queryAll

  public INDArray queryAll(INDArray toQuery,
                           int n)

  Query results up to length n nearest neighbors

  Parameters:

  toQuery - the query item

  n - the number of nearest neighbors for the given data point

  Returns:

  the indices for the nearest neighbors

• queryWithDistances

  public List<Pair<Double,Integer>> queryWithDistances(INDArray query,
                                                       int numResults)

  Query all with the distances sorted by index

  Parameters:

  query - the query vector

  numResults - the number of results to return

  Returns:

  a list of samples