rectify shuffle()

2018-07-19 09:03:03 +05:30
parent 3b7bbf0f14
commit 3bd4acfdf3
1 changed files with 2 additions and 0 deletions
@@ -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++)
  {