ra_query_stat.hpp

Go to the documentation of this file.

 
#ifndef MLPACK_METHODS_RANN_RA_QUERY_STAT_HPP
#define MLPACK_METHODS_RANN_RA_QUERY_STAT_HPP
 
#include <mlpack/prereqs.hpp>
 
#include <mlpack/core/tree/binary_space_tree.hpp>
 
#include <mlpack/core/metrics/lmetric.hpp>
#include <mlpack/methods/neighbor_search/sort_policies/nearest_neighbor_sort.hpp>
 
namespace mlpack {
namespace neighbor {
 
template<typename SortPolicy>
class RAQueryStat
{
 public:
  RAQueryStat() : bound(SortPolicy::WorstDistance()), numSamplesMade(0) { }
 
  template<typename TreeType>
  RAQueryStat(const TreeType& /* node */) :
    bound(SortPolicy::WorstDistance()),
    numSamplesMade(0)
  { }
 
  double Bound() const { return bound; }
  double& Bound() { return bound; }
 
  size_t NumSamplesMade() const { return numSamplesMade; }
  size_t& NumSamplesMade() { return numSamplesMade; }
 
  template<typename Archive>
  void serialize(Archive& ar, const unsigned int /* version */)
  {
    ar & BOOST_SERIALIZATION_NVP(bound);
    ar & BOOST_SERIALIZATION_NVP(numSamplesMade);
  }
 
 private:
  double bound;
  size_t numSamplesMade;
};
 
} // namespace neighbor
} // namespace mlpack
 
#endif