mlpack  1.0.12
range_search_rules.hpp
Go to the documentation of this file.
1 
14 #ifndef __MLPACK_METHODS_RANGE_SEARCH_RANGE_SEARCH_RULES_HPP
15 #define __MLPACK_METHODS_RANGE_SEARCH_RANGE_SEARCH_RULES_HPP
16 
17 #include "../neighbor_search/ns_traversal_info.hpp"
18 
19 namespace mlpack {
20 namespace range {
21 
22 
23 template<typename MetricType, typename TreeType>
25 {
26  public:
38  RangeSearchRules(const arma::mat& referenceSet,
39  const arma::mat& querySet,
40  const math::Range& range,
41  std::vector<std::vector<size_t> >& neighbors,
42  std::vector<std::vector<double> >& distances,
43  MetricType& metric);
44 
51  double BaseCase(const size_t queryIndex, const size_t referenceIndex);
52 
61  double Score(const size_t queryIndex, TreeType& referenceNode);
62 
74  double Rescore(const size_t queryIndex,
75  TreeType& referenceNode,
76  const double oldScore) const;
77 
86  double Score(TreeType& queryNode, TreeType& referenceNode);
87 
99  double Rescore(TreeType& queryNode,
100  TreeType& referenceNode,
101  const double oldScore) const;
102 
104 
105  const TraversalInfoType& TraversalInfo() const { return traversalInfo; }
106  TraversalInfoType& TraversalInfo() { return traversalInfo; }
107 
108  private:
110  const arma::mat& referenceSet;
111 
113  const arma::mat& querySet;
114 
117 
119  std::vector<std::vector<size_t> >& neighbors;
120 
122  std::vector<std::vector<double> >& distances;
123 
125  MetricType& metric;
126 
131 
135  void AddResult(const size_t queryIndex,
136  TreeType& referenceNode);
137 
138  TraversalInfoType traversalInfo;
139 };
140 
141 }; // namespace range
142 }; // namespace mlpack
143 
144 // Include implementation.
145 #include "range_search_rules_impl.hpp"
146 
147 #endif
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.
Definition: load.hpp:23
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
const math::Range & range
The range of distances for which we are searching.
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.
Definition: range.hpp:23
std::vector< std::vector< size_t > > & neighbors
The vector the resultant neighbor indices should be stored in.