random_partition.hpp

Go to the documentation of this file.

 
#ifndef MLPACK_METHODS_KMEANS_RANDOM_PARTITION_HPP
#define MLPACK_METHODS_KMEANS_RANDOM_PARTITION_HPP
 
#include <mlpack/prereqs.hpp>
 
namespace mlpack {
namespace kmeans {
 
class RandomPartition
{
 public:
  RandomPartition() { }
 
  template<typename MatType>
  inline static void Cluster(const MatType& data,
                             const size_t clusters,
                             arma::Row<size_t>& assignments)
  {
    // Implementation is so simple we'll put it here in the header file.
    assignments = arma::shuffle(arma::linspace<arma::Row<size_t>>(0,
        (clusters - 1), data.n_cols));
  }
 
  template<typename Archive>
  void serialize(Archive& /* ar */, const unsigned int /* version */) { }
};
 
} // namespace kmeans
} // namespace mlpack
 
#endif