14 #ifndef __MLPACK_METHODS_RANGE_SEARCH_RANGE_SEARCH_RULES_HPP
15 #define __MLPACK_METHODS_RANGE_SEARCH_RANGE_SEARCH_RULES_HPP
17 #include "../neighbor_search/ns_traversal_info.hpp"
23 template<
typename MetricType,
typename TreeType>
43 std::vector<std::vector<size_t> >&
neighbors,
44 std::vector<std::vector<double> >&
distances,
54 double BaseCase(
const size_t queryIndex,
const size_t referenceIndex);
64 double Score(
const size_t queryIndex, TreeType& referenceNode);
77 double Rescore(
const size_t queryIndex,
78 TreeType& referenceNode,
79 const double oldScore)
const;
89 double Score(TreeType& queryNode, TreeType& referenceNode);
102 double Rescore(TreeType& queryNode,
103 TreeType& referenceNode,
104 const double oldScore)
const;
147 TreeType& referenceNode);
161 #include "range_search_rules_impl.hpp"
const arma::mat & referenceSet
The reference set.
RangeSearchRules(const arma::mat &referenceSet, const arma::mat &querySet, const math::Range &range, std::vector< std::vector< size_t > > &neighbors, std::vector< std::vector< double > > &distances, MetricType &metric, const bool sameSet=false)
Construct the RangeSearchRules object.
size_t lastReferenceIndex
The last reference index.
void AddResult(const size_t queryIndex, TreeType &referenceNode)
Add all the points in the given node to the results for the given query point.
size_t BaseCases() const
Get the number of base cases.
Linear algebra utility functions, generally performed on matrices or vectors.
std::vector< std::vector< double > > & distances
The vector the resultant neighbor distances should be stored in.
double Score(const size_t queryIndex, TreeType &referenceNode)
Get the score for recursion order.
size_t lastQueryIndex
The last query index.
Traversal information for NeighborSearch.
size_t baseCases
The number of base cases.
const arma::mat & querySet
The query set.
neighbor::NeighborSearchTraversalInfo< TreeType > TraversalInfoType
const TraversalInfoType & TraversalInfo() const
Simple real-valued range.
bool sameSet
If true, the query and reference set are taken to be the same.
TraversalInfoType traversalInfo
const math::Range & range
The range of distances for which we are searching.
TraversalInfoType & TraversalInfo()
double BaseCase(const size_t queryIndex, const size_t referenceIndex)
Compute the base case between the given query point and reference point.
size_t scores
THe number of scores.
size_t Scores() const
Get the number of scores (that is, calls to RangeDistance()).
double Rescore(const size_t queryIndex, TreeType &referenceNode, const double oldScore) const
Re-evaluate the score for recursion order.
MetricType & metric
The instantiated metric.
std::vector< std::vector< size_t > > & neighbors
The vector the resultant neighbor indices should be stored in.