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);