当列表项失败或耗时太长时,如何超时?

时间:2012-06-08 16:32:26

标签: r multicore plyr

对于我目前参与的几项努力,我通过一系列功能运行具有众多参数组合的大型数据集。这些函数有一个包装器(所以我可以mclapply),以便于在集群上操作。但是,我遇到了两个主要挑战。

a)我的参数组合很大(想想20k到100k)。有时特定的组合失败(例如,生存太高而死亡率太低,因此模型永远不会收敛为假设情景)。我很难提前确定哪些组合会失败(如果我能做到这一点,生活会更容易)。但是现在我有这种类型的设置:

failsafe <- failwith(NULL, my_wrapper_function)
# This is what I run
# Note that input_variables contains a list of variables in each list item
results <-  mclapply(input_variables, failsafe, mc.cores = 72)
# On my local dual core mac, I can't do this so the equivalent would be:
results <-  llply(input_variables, failsafe,  .progress = 'text')

我的包装函数的骨架如下所示:

my_wrapper_function <- function(tlist) {
    run <- tryCatch(my_model(tlist$a, tlist$b, tlist$sA, tlist$Fec, m = NULL) , error=function(e) NULL)
...
return(run)
}

这是最有效的方法吗?如果由于某种原因,特定的变量组合会使模型崩溃,我需要它返回NULL并继续进行其余的操作。但是,我仍然有一些问题,这个问题的失败不如优雅。

b)有时输入的特定组合不会使模型崩溃,但需要很长时间才能收敛。我对我的集群上的计算时间设置了一个限制(比如6个小时),所以我不会浪费我的资源来解决问题。如何包含超时,如果函数调用在单个列表项上花费的时间超过x时间,它应该继续?计算花费的时间是微不足道的,但是模拟中间的函数不能被中断以检查时间,对吗?

感谢任何想法,解决方案或技巧!

1 个答案:

答案 0 :(得分:12)

您可以使用tryCatch()包中evalWithTimeout()R.utils的组合来管理优雅的退出时间。 See also this post,它提供了类似的代码,可以更详细地解压缩它。

require(R.utils)

myFun <- function(x) {Sys.sleep(x); x^2}

## evalWithTimeout() times out evaluation after 3.1 seconds, and then
## tryCatch() handles the resulting error (of class "TimeoutException") with 
## grace and aplomb.
myWrapperFunction <- function(i) {
    tryCatch(expr = evalWithTimeout(myFun(i), timeout = 3.1), 
             TimeoutException = function(ex) "TimedOut")
}

sapply(1:5, myWrapperFunction)
# [1] "1"        "4"        "9"        "TimedOut" "TimedOut"