drusilla_select.hpp

Go to the documentation of this file.

#ifndef MLPACK_METHODS_APPROX_KFN_DRUSILLA_SELECT_HPP
#define MLPACK_METHODS_APPROX_KFN_DRUSILLA_SELECT_HPP

#include <mlpack/prereqs.hpp>

namespace mlpack {
namespace neighbor {

template<typename MatType = arma::mat>
class DrusillaSelect
{
 public:
  DrusillaSelect(const MatType& referenceSet,
                 const size_t l,
                 const size_t m);

  DrusillaSelect(const size_t l, const size_t m);

  void Train(const MatType& referenceSet,
             const size_t l = 0,
             const size_t m = 0);

  void Search(const MatType& querySet,
              const size_t k,
              arma::Mat<size_t>& neighbors,
              arma::mat& distances);

  template<typename Archive>
  void Serialize(Archive& ar, const unsigned int /* version */);

  const MatType& CandidateSet() const { return candidateSet; }
  MatType& CandidateSet() { return candidateSet; }

  const arma::Col<size_t>& CandidateIndices() const { return candidateIndices; }
  arma::Col<size_t>& CandidateIndices() { return candidateIndices; }

 private:
  MatType candidateSet;
  arma::Col<size_t> candidateIndices;

  size_t l;
  size_t m;
};

} // namespace neighbor
} // namespace mlpack

// Include implementation.
#include "drusilla_select_impl.hpp"

#endif