diff --git a/src/mlpack/methods/kmeans/elkan_kmeans_impl.hpp b/src/mlpack/methods/kmeans/elkan_kmeans_impl.hpp index 638e35c873..c2682f7382 100644 --- a/src/mlpack/methods/kmeans/elkan_kmeans_impl.hpp +++ b/src/mlpack/methods/kmeans/elkan_kmeans_impl.hpp @@ -156,7 +156,7 @@ double ElkanKMeans::Iterate(const arma::mat& centroids, if (counts[c] > 0) newCentroids.col(c) /= counts[c]; else - newCentroids.fill(DBL_MAX); // Fill with invalid value. + newCentroids.col(c).fill(DBL_MAX); // Fill with invalid value. moveDistances(c) = metric.Evaluate(newCentroids.col(c), centroids.col(c)); cNorm += std::pow(moveDistances(c), 2.0);