allow_empty_clusters.hpp

Go to the documentation of this file.

 
#ifndef MLPACK_METHODS_KMEANS_ALLOW_EMPTY_CLUSTERS_HPP
#define MLPACK_METHODS_KMEANS_ALLOW_EMPTY_CLUSTERS_HPP
 
#include <mlpack/prereqs.hpp>
 
namespace mlpack {
namespace kmeans {
 
class AllowEmptyClusters
{
 public:
  AllowEmptyClusters() { }
 
  template<typename MetricType, typename MatType>
  static inline force_inline void EmptyCluster(
      const MatType& /* data */,
      const size_t emptyCluster,
      const arma::mat& oldCentroids,
      arma::mat& newCentroids,
      arma::Col<size_t>& /* clusterCounts */,
      MetricType& /* metric */,
      const size_t /* iteration */)
  {
    // Take the last iteration's centroid.
    newCentroids.col(emptyCluster) = oldCentroids.col(emptyCluster);
  }
 
  template<typename Archive>
  void serialize(Archive& /* ar */, const unsigned int /* version */) { }
};
 
} // namespace kmeans
} // namespace mlpack
 
#endif