diff --git a/src/mlpack/methods/kmeans/elkan_kmeans_impl.hpp b/src/mlpack/methods/kmeans/elkan_kmeans_impl.hpp index 66eed485ff..6602d8e5bb 100644 --- a/src/mlpack/methods/kmeans/elkan_kmeans_impl.hpp +++ b/src/mlpack/methods/kmeans/elkan_kmeans_impl.hpp @@ -88,8 +88,10 @@ double ElkanKMeans::Iterate(const arma::mat& centroids, } // Create thread-local storage for newCentroids and counts - std::vector threadNewCentroids(numThreads, arma::mat(centroids.n_rows, centroids.n_cols, arma::fill::zeros)); - std::vector> threadCounts(numThreads, arma::Col(centroids.n_cols, arma::fill::zeros)); + std::vector threadNewCentroids(numThreads, + arma::mat(centroids.n_rows, centroids.n_cols, arma::fill::zeros)); + std::vector> threadCounts(numThreads, + arma::Col(centroids.n_cols, arma::fill::zeros)); // Now loop over all points, and see which ones need to be updated. #pragma omp parallel for schedule(dynamic) reduction(+:distanceCalculations) @@ -103,7 +105,8 @@ double ElkanKMeans::Iterate(const arma::mat& centroids, { // No change needed. This point must still belong to that cluster. threadCounts[threadId](assignments[i])++; - threadNewCentroids[threadId].col(assignments[i]) += arma::vec(dataset.col(i)); + threadNewCentroids[threadId].col(assignments[i]) += + arma::vec(dataset.col(i)); continue; } else @@ -166,7 +169,8 @@ double ElkanKMeans::Iterate(const arma::mat& centroids, // At this point, we know the new cluster assignment. // Step 4: for each center c, let m(c) be the mean of the points assigned to // c. - threadNewCentroids[threadId].col(assignments[i]) += arma::vec(dataset.col(i)); + threadNewCentroids[threadId].col(assignments[i]) += + arma::vec(dataset.col(i)); threadCounts[threadId][assignments[i]]++; }