Question

是否可以使用load()？

为加载到R中的数据创建进度条

对于数据分析项目，大型矩阵从.RData文件加载到R中，这需要几分钟才能加载。我想有一个进度条来监控数据加载前的时间。 R已集成了良好的progress bar功能，但load（）没有用于监视已读取数据量的挂钩。如果我不能直接使用负载，是否有间接方式可以创建这样的进度条？也许将.RData文件加载到chucks并将它们放在一起用于R.是否有人对此有任何想法或建议？

Answer 1

我提出了以下解决方案，该解决方案适用于小于2 ^ 32 - 1字节的文件大小。

R对象需要序列化并保存到文件中，如下面的代码所示。

saveObj <- function(object, file.name){
    outfile <- file(file.name, "wb")
    serialize(object, outfile)
    close(outfile)
}

然后我们以块的形式读取二进制数据，跟踪读取的数量并相应地更新进度条。

loadObj <- function(file.name){
    library(foreach)
    filesize <- file.info(file.name)$size
    chunksize <- ceiling(filesize / 100)
    pb <- txtProgressBar(min = 0, max = 100, style=3)
    infile <- file(file.name, "rb")
    data <- foreach(it = icount(100), .combine = c) %do% {
        setTxtProgressBar(pb, it)
        readBin(infile, "raw", chunksize)
    }
    close(infile)
    close(pb)
    return(unserialize(data))
}

代码可以按如下方式运行：

> a <- 1:100000000
> saveObj(a, "temp.RData")
> b <- loadObj("temp.RData")
  |======================================================================| 100%
> all.equal(b, a)
[1] TRUE

如果我们对进度条方法进行基准测试而不是在单个块中读取文件，我们会看到进度条方法稍慢，但不足以担心。

> system.time(unserialize(readBin(infile, "raw", file.info("temp.RData")$size)))
   user  system elapsed
  2.710   0.340   3.062
> system.time(b <- loadObj("temp.RData"))
  |======================================================================| 100%
   user  system elapsed
  3.750   0.400   4.154

因此，虽然上述方法有效，但由于文件大小的限制，我觉得它完全没用。进度条仅适用于需要很长时间才能读入的大型文件。

如果有人能提出比这个解决方案更好的东西，那将会很棒！

Answer 2

我可能会建议加快加载（并保存）时间，以便不需要进度条吗？如果读取一个矩阵是“快速”，那么您可能会报告每个读取矩阵之间的进度（如果您有很多）。

这是一些测量。通过简单地设置compress = FALSE，加载速度加倍。但是通过编写一个简单的矩阵串行器，加载速度几乎快了20倍。

x <- matrix(runif(1e7), 1e5) # Matrix with 100k rows and 100 columns

system.time( save('x', file='c:/foo.bin') ) # 13.26 seconds
system.time( load(file='c:/foo.bin') ) # 2.03 seconds

system.time( save('x', file='c:/foo.bin', compress=FALSE) ) # 0.86 seconds
system.time( load(file='c:/foo.bin') ) # 0.92 seconds

system.time( saveMatrix(x, 'c:/foo.bin') ) # 0.70 seconds
system.time( y <- loadMatrix('c:/foo.bin') ) # 0.11 seconds !!!
identical(x,y)

saveMatrix / loadMatrix的定义如下。它们目前不处理dimnames和其他属性，但可以轻松添加。

saveMatrix <- function(m, fileName) {
    con <- file(fileName, 'wb')
    on.exit(close(con))
    writeBin(dim(m), con)
    writeBin(typeof(m), con)
    writeBin(c(m), con)
}

loadMatrix <- function(fileName) {
    con <- file(fileName, 'rb')
    on.exit(close(con))
    d <- readBin(con, 'integer', 2)
    type <- readBin(con, 'character', 1)
    structure(readBin(con, type, prod(d)), dim=d)
}

如何为R中的数据加载创建进度条？

2 个答案: