qdafn.hpp

Go to the documentation of this file.

#ifndef MLPACK_METHODS_APPROX_KFN_QDAFN_HPP
#define MLPACK_METHODS_APPROX_KFN_QDAFN_HPP

#include <mlpack/prereqs.hpp>
#include <mlpack/core/dists/gaussian_distribution.hpp>

namespace mlpack {
namespace neighbor {

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

  QDAFN(const MatType& referenceSet,
        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 uint32_t /* version */);

  size_t NumProjections() const { return candidateSet.size(); }

  const MatType& CandidateSet(const size_t t) const { return candidateSet[t]; }
  MatType& CandidateSet(const size_t t) { return candidateSet[t]; }

 private:
  size_t l;
  size_t m;
  arma::mat lines;
  arma::mat projections;

  arma::Mat<size_t> sIndices;
  arma::mat sValues;

  // Candidate sets; one element in the vector for each table.
  std::vector<MatType> candidateSet;
};

} // namespace neighbor
} // namespace mlpack

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

#endif