kill_empty_clusters.hpp

Go to the documentation of this file.

 
#ifndef __MLPACK_METHODS_KMEANS_KILL_EMPTY_CLUSTERS_HPP
#define __MLPACK_METHODS_KMEANS_KILL_EMPTY_CLUSTERS_HPP
 
#include <mlpack/prereqs.hpp>
 
namespace mlpack {
namespace kmeans {
 
class KillEmptyClusters
{
 public:
  KillEmptyClusters() { }
 
  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 */)
{
  // Remove the empty cluster.
  if (emptyCluster < newCentroids.n_cols)
  {
    newCentroids.shed_col(emptyCluster);
    clusterCounts.shed_row(emptyCluster);
  }
}
 
  template<typename Archive>
  void serialize(Archive& /* ar */, const unsigned int /* version */) { }
};
 
} // namespace kmeans
} // namespace mlpack
 
#endif