diff --git a/src/mlpack/methods/lmnn/lmnn_function_impl.hpp b/src/mlpack/methods/lmnn/lmnn_function_impl.hpp index c3974e5746..469c979d8c 100644 --- a/src/mlpack/methods/lmnn/lmnn_function_impl.hpp +++ b/src/mlpack/methods/lmnn/lmnn_function_impl.hpp @@ -77,6 +77,7 @@ void LMNNFunction::Shuffle() arma::cube newEvalOld = evalOld; arma::vec newlastTransformationIndices = lastTransformationIndices; arma::mat newMaxImpNorm = maxImpNorm; + arma::vec newNorm = norm; // Generate ordering. arma::uvec ordering = arma::shuffle(arma::linspace(0, @@ -89,6 +90,7 @@ void LMNNFunction::Shuffle() labels = newLabels.cols(ordering); maxImpNorm = newMaxImpNorm.cols(ordering); lastTransformationIndices = newlastTransformationIndices.elem(ordering); + norm = newNorm.elem(ordering); for (size_t i = 0; i < ordering.n_elem; i++) {