hamerly_kmeans.hpp

Go to the documentation of this file.

#ifndef MLPACK_METHODS_KMEANS_HAMERLY_KMEANS_HPP
#define MLPACK_METHODS_KMEANS_HAMERLY_KMEANS_HPP

namespace mlpack {
namespace kmeans {

template<typename MetricType, typename MatType>
class HamerlyKMeans
{
 public:
  HamerlyKMeans(const MatType& dataset, MetricType& metric);

  double Iterate(const arma::mat& centroids,
                 arma::mat& newCentroids,
                 arma::Col<size_t>& counts);

  size_t DistanceCalculations() const { return distanceCalculations; }

 private:
  const MatType& dataset;
  MetricType& metric;

  arma::vec minClusterDistances;

  arma::vec upperBounds;
  arma::vec lowerBounds;
  arma::Col<size_t> assignments;

  size_t distanceCalculations;
};

} // namespace kmeans
} // namespace mlpack

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

#endif