1

我正在尝试使用 snow 包在 R 中对弹性网络模型进行评分，但我无法弄清楚如何让预测函数在集群中的多个节点上运行。下面的代码包含时序基准和产生错误的实际代码：

##############
#Snow example#
##############

library(snow)
library(glmnet)
library(mlbench)

data(BostonHousing)
BostonHousing$chas<-as.numeric(BostonHousing$chas)

ind<-as.matrix(BostonHousing[,1:13],col.names=TRUE)
dep<-as.matrix(BostonHousing[,14],col.names=TRUE)

fit_lambda<-cv.glmnet(ind,dep)

#fit elastic net
fit_en<<-glmnet(ind,dep,family="gaussian",alpha=0.5,lambda=fit_lambda$lambda.min)

ind_exp<-rbind(ind,ind)

#single thread baseline
i<-0
while(i < 2000){
    ind_exp<-rbind(ind_exp,ind)
    i = i+1
    }

system.time(st<-predict(fit_en,ind_exp))

#formula for parallel execution
pred_en<-function(x){
    x<-as.matrix(x)
    return(predict(fit_en,x))
    }

#make the cluster
cl<-makeSOCKcluster(4)
clusterExport(cl,"fit_en")
clusterExport(cl,"pred_en")

#parallel baseline
system.time(mt<-parRapply(cl,ind_exp,pred_en))

我已经能够通过使用多核的 Linux 机器上的分叉进行并行化，但我最终不得不使用性能相当差的 mclapply 与 unlist 相结合，并且正在寻找一种更好的方法来处理雪（顺便说一句，这对我的dev windows PC 和我的 prod Linux 服务器）。谢谢。

4

1 回答 1

2

我应该首先说该predict.glmnet函数的计算密集度似乎不足以值得并行化。但这是一个有趣的例子，我的回答可能对你有帮助，即使这种特殊情况不值得并行化。

主要问题是该parRapply函数是一个并行包装器apply，它反过来在子矩阵的行上调用您的函数，这不是您想要的。您希望直接在子矩阵上调用您的函数。Snow 不包含执行此操作的便捷函数，但很容易编写：

rowchunkapply <- function(cl, x, fun, ...) {
    do.call('rbind', clusterApply(cl, splitRows(x, length(cl)), fun, ...))
}

您的示例中的另一个问题是您需要在工作人员上加载 glmnet 以便调用正确的预测函数。您也不需要显式导出pred_en函数，因为这是为您处理的。

这是我的示例版本：

library(snow)
library(glmnet)
library(mlbench)

data(BostonHousing)
BostonHousing$chas <- as.numeric(BostonHousing$chas)
ind <- as.matrix(BostonHousing[,1:13], col.names=TRUE)
dep <- as.matrix(BostonHousing[,14], col.names=TRUE)
fit_lambda <- cv.glmnet(ind, dep)
fit_en <- glmnet(ind, dep, family="gaussian", alpha=0.5,
                 lambda=fit_lambda$lambda.min)
ind_exp <- do.call("rbind", rep(list(ind), 2002))

# make and initialize the cluster
cl <- makeSOCKcluster(4)
clusterEvalQ(cl, library(glmnet))
clusterExport(cl, "fit_en")

# execute a function on row chunks of x and rbind the results
rowchunkapply <- function(cl, x, fun, ...) {
    do.call('rbind', clusterApply(cl, splitRows(x, length(cl)), fun, ...))
}

# worker function
pred_en <- function(x) {
    predict(fit_en, x)
}
mt <- rowchunkapply(cl, ind_exp, pred_en)

您可能也有兴趣使用cv.glmnet parallel使用 foreach 包的选项。

于 2014-02-06T02:08:31.683 回答