Kmeans.hpp

Go to the documentation of this file.

#pragma once
 
#include <Eigen/Dense>
#include <mutable/mutable.hpp>
#include <vector>
 
 
namespace m {
 
using MatrixRXf = Eigen::Matrix<float, Eigen::Dynamic, Eigen::Dynamic, Eigen::RowMajor>;
 
MatrixRXf M_EXPORT kmeans_plus_plus(const Eigen::MatrixXf &data, unsigned k);
 
std::pair<std::vector<unsigned>, MatrixRXf> M_EXPORT kmeans_with_centroids(const Eigen::MatrixXf &data, unsigned k);
 
inline std::vector<unsigned> M_EXPORT kmeans(const Eigen::MatrixXf &data, unsigned k) {
    return kmeans_with_centroids(data, k).first;
}
 
}