1 files changed, 4 insertions, 0 deletions
diff --git a/lamp.cpp b/lamp.cpp
index 7d36877..293ce30 100644
--- a/lamp.cpp
+++ b/lamp.cpp
@@ -54,4 +54,8 @@ void mp::lamp(const arma::mat &X, const arma::uvec &sampleIndices, const arma::m
 
         Y(i, arma::span(0, 1)) = (point - Xtil) * M + Ytil;
     }
+
+    for (arma::uword i = 0; i < sampleSize; i++) {
+        Y.row(sampleIndices[i]) = Ys.row(i);
+    }
 }