|
mlpack
2.0.1
|
This is an implementation of a single iteration of Lloyd's algorithm for k-means. More...
Public Member Functions | |
| NaiveKMeans (const MatType &dataset, MetricType &metric) | |
| Construct the NaiveKMeans object with the given dataset and metric. More... | |
| size_t | DistanceCalculations () const |
| 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 matrix. More... | |
Private Attributes | |
| const MatType & | dataset |
| The dataset. More... | |
| size_t | distanceCalculations |
| Number of distance calculations. More... | |
| MetricType & | metric |
| The instantiated metric. More... | |
This is an implementation of a single iteration of Lloyd's algorithm for k-means.
If your intention is to run the full k-means algorithm, you are looking for the mlpack::kmeans::KMeans class instead of this one. This class is used by KMeans as the actual implementation of the Lloyd iteration.
| MetricType | Type of metric used with this implementation. |
| MatType | Matrix type (arma::mat or arma::sp_mat). |
Definition at line 32 of file naive_kmeans.hpp.
| mlpack::kmeans::NaiveKMeans< MetricType, MatType >::NaiveKMeans | ( | const MatType & | dataset, |
| MetricType & | metric | ||
| ) |
Construct the NaiveKMeans object with the given dataset and metric.
| dataset | Dataset. |
| metric | Instantiated metric. |
|
inline |
Definition at line 54 of file naive_kmeans.hpp.
References mlpack::kmeans::NaiveKMeans< MetricType, MatType >::distanceCalculations.
| double mlpack::kmeans::NaiveKMeans< MetricType, MatType >::Iterate | ( | const arma::mat & | centroids, |
| arma::mat & | newCentroids, | ||
| arma::Col< size_t > & | counts | ||
| ) |
Run a single iteration of the Lloyd algorithm, updating the given centroids into the newCentroids matrix.
| centroids | Current cluster centroids. |
| newCentroids | New cluster centroids. |
|
private |
The dataset.
Definition at line 58 of file naive_kmeans.hpp.
|
private |
Number of distance calculations.
Definition at line 63 of file naive_kmeans.hpp.
Referenced by mlpack::kmeans::NaiveKMeans< MetricType, MatType >::DistanceCalculations().
|
private |
The instantiated metric.
Definition at line 60 of file naive_kmeans.hpp.
1.8.11