[mlpack-git] master: Minor speedup. (ee55885)
gitdub at big.cc.gt.atl.ga.us
gitdub at big.cc.gt.atl.ga.us
Wed Apr 29 14:43:43 EDT 2015
Repository : https://github.com/mlpack/mlpack
On branch : master
Link : https://github.com/mlpack/mlpack/compare/ee384655c4462e422e343e9725437fd772ca4449...182d4a629c1b23f683dff7b284844e4e3e9f5cc4
>---------------------------------------------------------------
commit ee55885c4f97a04172102114182ec556bdc94b91
Author: Ryan Curtin <ryan at ratml.org>
Date: Fri Mar 27 18:38:45 2015 +0000
Minor speedup.
>---------------------------------------------------------------
ee55885c4f97a04172102114182ec556bdc94b91
src/mlpack/methods/mean_shift/mean_shift_impl.hpp | 5 +++--
1 file changed, 3 insertions(+), 2 deletions(-)
diff --git a/src/mlpack/methods/mean_shift/mean_shift_impl.hpp b/src/mlpack/methods/mean_shift/mean_shift_impl.hpp
index 040a115..086961b 100644
--- a/src/mlpack/methods/mean_shift/mean_shift_impl.hpp
+++ b/src/mlpack/methods/mean_shift/mean_shift_impl.hpp
@@ -142,8 +142,9 @@ inline void MeanShift<KernelType, MatType>::Cluster(
bool isDuplicated = false;
for (size_t k = 0; k < centroids.n_cols; ++k)
{
- arma::Col<double> delta = allCentroids.col(i) - centroids.col(k);
- if (norm(delta, 2) < radius)
+ const double distance = metric::EuclideanDistance::Evaluate(
+ allCentroids.col(i), centroids.col(k));
+ if (distance < radius)
{
isDuplicated = true;
assignments(i) = k;
More information about the mlpack-git
mailing list