15#ifndef MLPACK_METHODS_KMEANS_NAIVE_KMEANS_HPP
16#define MLPACK_METHODS_KMEANS_NAIVE_KMEANS_HPP
31template<
typename MetricType,
typename MatType>
53 double Iterate(
const arma::mat& centroids,
54 arma::mat& newCentroids,
55 arma::Col<size_t>& counts);
61 const MatType& dataset;
66 size_t distanceCalculations;
73#include "naive_kmeans_impl.hpp"
This is an implementation of a single iteration of Lloyd's algorithm for k-means.
double Iterate(const arma::mat ¢roids, arma::mat &newCentroids, arma::Col< size_t > &counts)
Run a single iteration of the Lloyd algorithm, updating the given centroids into the newCentroids mat...
NaiveKMeans(const MatType &dataset, MetricType &metric)
Construct the NaiveKMeans object with the given dataset and metric.
size_t DistanceCalculations() const
Linear algebra utility functions, generally performed on matrices or vectors.
The core includes that mlpack expects; standard C++ includes and Armadillo.