From 3bd4acfdf3c33c249952a29b75ca4e7ae13f4064 Mon Sep 17 00:00:00 2001
From: Manish <manish887kr@gmail.com>
Date: Thu, 19 Jul 2018 09:03:03 +0530
Subject: [PATCH] rectify shuffle()

---
 src/mlpack/methods/lmnn/lmnn_function_impl.hpp | 2 ++
 1 file changed, 2 insertions(+)

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<MetricType>::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<arma::uvec>(0,
@@ -89,6 +90,7 @@ void LMNNFunction<MetricType>::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++)
   {