Question

我正在尝试从数据框创建稀疏数据矩阵，而不必构建密集矩阵，这会导致严重的内存问题。

我发现以下帖子似乎找到了一个解决方案： Create Sparse Matrix from a data frame

我已经尝试过这个解决方案了，但是，它对我不起作用，可能是因为我的UserID和MovieID并没有从1开始。

以下是我的示例代码：

library(Matrix)

UserID<-c(10090,10090,10090,10316,10316)
MovieID <-c(63155,63530,63544,63155,63545)
Rating <-c(2,2,1,2,1)
trainingData<-data.frame(UserID,MovieID,Rating)
trainingData

UIMatrix <- sparseMatrix(i = trainingData$UserID,
                         j = trainingData$MovieID,
                         x = trainingData$Rating)

dim(UIMatrix)

我希望得到一个2 x 3矩阵，但dims对应于最大用户和电影ID。

我已经尝试了帖子中提出的第二个解决方案，但它并不能解决数据的问题。

有人可以提供一些建议吗？

Answer 1

您可以将索引转换为从as.integer(as.factor(.))开始的索引。

UIMatrix <- sparseMatrix(i = as.integer(as.factor(trainingData$UserID)),
                         j = as.integer(as.factor(trainingData$MovieID)),
                         x = trainingData$Rating)

dim(UIMatrix)
# [1] 2 4

dimnames(UIMatrix) <- list(sort(unique(trainingData$UserID)),
                           sort(unique(trainingData$MovieID)))

UIMatrix
# 2 x 4 sparse Matrix of class "dgCMatrix"
#       63155 63530 63544 63545
# 10090     2     2     1     .
# 10316     2     .     .     1

从数据框创建稀疏矩阵

1 个答案: