pelleg_moore_kmeans_rules.hpp

Go to the documentation of this file.

#ifndef MLPACK_METHODS_KMEANS_PELLEG_MOORE_KMEANS_RULES_HPP
#define MLPACK_METHODS_KMEANS_PELLEG_MOORE_KMEANS_RULES_HPP

namespace mlpack {
namespace kmeans {

template<typename MetricType, typename TreeType>
class PellegMooreKMeansRules
{
 public:
  PellegMooreKMeansRules(const typename TreeType::Mat& dataset,
                         const arma::mat& centroids,
                         arma::mat& newCentroids,
                         arma::Col<size_t>& counts,
                         MetricType& metric);

  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);

  size_t DistanceCalculations() const { return distanceCalculations; }
  size_t& DistanceCalculations() { return distanceCalculations; }

 private:
  const typename TreeType::Mat& dataset;
  const arma::mat& centroids;
  arma::mat& newCentroids;
  arma::Col<size_t>& counts;
  MetricType& metric;

  size_t distanceCalculations;
};

} // namespace kmeans
} // namespace mlpack

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

#endif