kde_rules.hpp

Go to the documentation of this file.

#ifndef MLPACK_METHODS_KDE_RULES_HPP
#define MLPACK_METHODS_KDE_RULES_HPP

#include <mlpack/core/tree/traversal_info.hpp>

namespace mlpack {
namespace kde {

template<typename MetricType, typename KernelType, typename TreeType>
class KDERules
{
 public:
  KDERules(const arma::mat& referenceSet,
           const arma::mat& querySet,
           arma::vec& densities,
           const double relError,
           const double absError,
           MetricType& metric,
           KernelType& kernel,
           const bool sameSet);

  double BaseCase(const size_t queryIndex, const size_t referenceIndex);

  double Score(const size_t queryIndex, TreeType& referenceNode);

  double Rescore(const size_t queryIndex,
                 TreeType& referenceNode,
                 const double oldScore) const;

  double Score(TreeType& queryNode, TreeType& referenceNode);

  double Rescore(TreeType& queryNode,
                 TreeType& referenceNode,
                 const double oldScore) const;

  typedef typename tree::TraversalInfo<TreeType> TraversalInfoType;

  const TraversalInfoType& TraversalInfo() const { return traversalInfo; }

  TraversalInfoType& TraversalInfo() { return traversalInfo; }

  size_t BaseCases() const { return baseCases; }

  size_t Scores() const { return scores; }

 private:
  double EvaluateKernel(const size_t queryIndex,
                        const size_t referenceIndex) const;

  double EvaluateKernel(const arma::vec& query,
                        const arma::vec& reference) const;

  const arma::mat& referenceSet;

  const arma::mat& querySet;

  arma::vec& densities;

  const double absError;

  const double relError;

  MetricType& metric;

  KernelType& kernel;

  const bool sameSet;

  size_t lastQueryIndex;

  size_t lastReferenceIndex;

  TraversalInfoType traversalInfo;

  size_t baseCases;

  size_t scores;
};

} // namespace kde
} // namespace mlpack

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

#endif