矩阵乘法和加法的R函数

Question

在矩阵上执行SVD后，我想创建一个函数(我不擅长使用R中的函数)，它根据我指定的n值创建一个简化矩阵。

例如，这是我现在的R代码。

scores = c(3,1,1,-1,3,1)
Mat = matrix(scores, nrow=2)
svd = svd(Mat)

现在，从矩阵分解到原始矩阵，通过蛮力和无知，下面的工作。

score1 = svd$u[,1] %*% t(svd$v[,1]) * svd$d[1]
    score2 = svd$u[,2] %*% t(svd$v[,2]) * svd$d[2]
z = score1 + score2
z
    [,1] [,2] [,3]
[1,]    3    1    3
[2,]    1   -1    1

接下来，我希望能够在一个大矩阵上做到这一点，并且希望能够指定因子的数量。所以，我想用一个函数来代替求和分数1:n

Answer 1

你不能把你的矩阵算法封装在你自己的一个小函数里吗?

recover_matrix_from_svd <- function(svd) {
    score <- 0
    for(i in 1:ncol(svd$u)) {
    score <- score + svd$u[,i] %*% t(svd$v[,i]) * svd$d[i]
    }
    score
}

或者，diag函数对此非常有用。使用它可以使计算更加清晰:

recover_matrix_from_svd <- function(svd) {
    svd$u %*% diag(svd$d) %*% t(svd$v)
}