InitializeRandom.hpp

Go to the documentation of this file.

#ifndef KMEANS_INITIALIZE_RANDOM_HPP
#define KMEANS_INITIALIZE_RANDOM_HPP 
 
#include <algorithm>
#include <cstdint>
#include <random>
 
#include "Initialize.hpp"
#include "copy_into_array.hpp"
 
#include "aarand/aarand.hpp"
 
namespace kmeans {
 
struct InitializeRandomOptions {
    uint64_t seed = 6523u;
};
 
template<typename Index_, typename Data_, typename Cluster_, typename Float_, class Matrix_ = Matrix<Index_, Data_> >
class InitializeRandom final : public Initialize<Index_, Data_, Cluster_, Float_, Matrix_> { 
private:
    InitializeRandomOptions my_options;
 
public:
    InitializeRandom(InitializeRandomOptions options) : my_options(std::move(options)) {}
 
    InitializeRandom() = default;
 
public:
    InitializeRandomOptions& get_options() {
        return my_options;
    }
 
public:
    Cluster_ run(const Matrix_& data, Cluster_ ncenters, Float_* centers) const {
        std::mt19937_64 eng(my_options.seed);
        auto nobs = data.num_observations();
        std::vector<Index_> chosen(std::min(nobs, static_cast<Index_>(ncenters)));
        aarand::sample(nobs, ncenters, chosen.begin(), eng);
        internal::copy_into_array(data, chosen, centers);
        return chosen.size();
    }
};
 
}
 
#endif