12 #ifndef MLPACK_METHODS_RANGE_SEARCH_RANGE_SEARCH_RULES_HPP
13 #define MLPACK_METHODS_RANGE_SEARCH_RANGE_SEARCH_RULES_HPP
27 template<
typename MetricType,
typename TreeType>
45 const arma::mat& querySet,
47 std::vector<std::vector<size_t> >& neighbors,
48 std::vector<std::vector<double> >& distances,
50 const bool sameSet =
false);
58 double BaseCase(
const size_t queryIndex,
const size_t referenceIndex);
68 double Score(
const size_t queryIndex, TreeType& referenceNode);
81 double Rescore(
const size_t queryIndex,
82 TreeType& referenceNode,
83 const double oldScore)
const;
93 double Score(TreeType& queryNode, TreeType& referenceNode);
106 double Rescore(TreeType& queryNode,
107 TreeType& referenceNode,
108 const double oldScore)
const;
122 const arma::mat& referenceSet;
125 const arma::mat& querySet;
131 std::vector<std::vector<size_t> >& neighbors;
134 std::vector<std::vector<double> >& distances;
143 size_t lastQueryIndex;
145 size_t lastReferenceIndex;
150 void AddResult(
const size_t queryIndex,
151 TreeType& referenceNode);
165 #include "range_search_rules_impl.hpp"
const TraversalInfoType & TraversalInfo() const
The TraversalInfo class holds traversal information which is used in dual-tree (and single-tree) trav...
size_t Scores() const
Get the number of scores (that is, calls to RangeDistance()).
double Score(const size_t queryIndex, TreeType &referenceNode)
Get the score for recursion order.
double BaseCase(const size_t queryIndex, const size_t referenceIndex)
Compute the base case between the given query point and reference point.
The RangeSearchRules class is a template helper class used by RangeSearch class when performing range...
TraversalInfoType & TraversalInfo()
size_t BaseCases() const
Get the number of base cases.
tree::TraversalInfo< TreeType > TraversalInfoType
double Rescore(const size_t queryIndex, TreeType &referenceNode, const double oldScore) const
Re-evaluate the score for recursion order.
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.