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>
41 std::vector<std::vector<size_t> >&
neighbors,
42 std::vector<std::vector<double> >&
distances,
51 double BaseCase(
const size_t queryIndex,
const size_t referenceIndex);
61 double Score(
const size_t queryIndex, TreeType& referenceNode);
74 double Rescore(
const size_t queryIndex,
75 TreeType& referenceNode,
76 const double oldScore)
const;
86 double Score(TreeType& queryNode, TreeType& referenceNode);
99 double Rescore(TreeType& queryNode,
100 TreeType& referenceNode,
101 const double oldScore)
const;
136 TreeType& referenceNode);
145 #include "range_search_rules_impl.hpp"
const arma::mat & referenceSet
The reference set.
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.
Linear algebra utility functions, generally performed on matrices or vectors.
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)
Construct the RangeSearchRules object.
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.
const arma::mat & querySet
The query set.
neighbor::NeighborSearchTraversalInfo< TreeType > TraversalInfoType
const TraversalInfoType & TraversalInfo() const
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.
double Rescore(const size_t queryIndex, TreeType &referenceNode, const double oldScore) const
Re-evaluate the score for recursion order.
MetricType & metric
The instantiated metric.
Simple real-valued range.
std::vector< std::vector< size_t > > & neighbors
The vector the resultant neighbor indices should be stored in.