From e6616bd33367d8ecbfbcf385535ce190bc386912 Mon Sep 17 00:00:00 2001
From: AdarshSantoria <adarshsantoria@gmail.com>
Date: Tue, 14 Feb 2023 02:29:12 +0530
Subject: [PATCH] fix bksvd method

---
 .../cf/decomposition_policies/block_krylov_svd_method.hpp     | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/mlpack/methods/cf/decomposition_policies/block_krylov_svd_method.hpp b/src/mlpack/methods/cf/decomposition_policies/block_krylov_svd_method.hpp
index 7b721e8ff4..fdc367d54f 100644
--- a/src/mlpack/methods/cf/decomposition_policies/block_krylov_svd_method.hpp
+++ b/src/mlpack/methods/cf/decomposition_policies/block_krylov_svd_method.hpp
@@ -76,9 +76,9 @@ class BlockKrylovSVDPolicy
     // Do singular value decomposition using the block krylov SVD algorithm.
     RandomizedBlockKrylovSVD blockkrylovsvd;
     blockkrylovsvd.Apply(data, w, sigma, h, rank);
-    
+
     // Sigma matrix is multiplied to w.
-    w = w * sigma;
+    w = w * arma::diagmat(sigma);
 
     // Take transpose of the matrix h as required by CF class.
     h = arma::trans(h);