parLapply 的實務應用

R 平行程式設計

Nabeel Imam

Data Scientist

來認識 workers

cluster <- makeCluster(4)

clusterEvalQ(cluster, {
  id <- Sys.getpid()
  print(
    paste("Hello, my worker ID is", id)
  )
})

[[1]]
[1] "Hello, my worker ID is 425108"

[[2]]
[1] "Hello, my worker ID is 425129"

[[3]]
[1] "Hello, my worker ID is 425150"

[[4]]
[1] "Hello, my worker ID is 425171"

平行篩選資料

print(file_list)

 [1] "./health/Afghanistan.csv"            
 [2] "./health/Albania.csv"            
 [3] "./health/Algeria.csv"            
 [4] "./health/American Samoa.csv"     
 [5] "./health/Andorra.csv"            
...

聽診器放在一疊百元美鈔上。

平行篩選資料

filterCSV <- function (csv) {
  read.csv(csv) %>% 
    dplyr::filter(!is.na(health_exp_pc))
}


cl <- makeCluster(4)
ls_df <- parLapply(cl, file_list, filterCSV)
stopCluster(cl)

Error in checkForRemoteErrors(val) :
  first error: could not find function "%>%"

clusterEvalQ 來救援

在叢集上載入套件

cl <- makeCluster(4)
clusterEvalQ(cl, library(dplyr))


ls_df <- parLapply(cl, file_list, filterCSV)
stopCluster(cl)

在叢集上載入多個套件

clusterEvalQ(cl, {
  library(dplyr)
  library(stringr)
})

[[1]]
       Country health_exp_pc Year
1  Afghanistan      81.27103 2002
2  Afghanistan      82.45785 2003
3  Afghanistan      89.47005 2004
...

[[2]]
   Country health_exp_pc Year
1  Albania      300.2757 2001
2  Albania      314.3254 2002
3  Albania      343.9442 2003
...

條件式篩選

# Function with an argument for starting year
filterCSV <- function (csv, min_year) { 

  read.csv(csv) %>% 
    dplyr::filter(!is.na(health_exp_pc),

                  # Filter data for min_year and onwards
                  Year >= min_year) 
}


selected_year <- 2010 # Value to be supplied to min_year

條件式篩選

cl <- makeCluster(4)
clusterEvalQ(cl, library(dplyr))

clusterExport(cl, "selected_year",

             envir = environment())


ls_df <- parLapply(cl, file_list, filterCSV,

                   min_year = selected_year)

stopCluster(cl)

將 selected_year 匯出到叢集
從目前環境匯出

將 selected_year 提供給 min_year

條件式篩選

[[1]]
       Country health_exp_pc Year
1  Afghanistan      143.6695 2010
2  Afghanistan      143.0915 2011
3  Afghanistan      151.9180 2012
...
  Country health_exp_pc Year
1 Albania      451.8820 2010
2 Albania      485.5835 2011
3 Albania      529.6322 2012
...

叢集衛生檢查清單

決定核心數
建立合適的叢集
- PSOCK：相容所有系統
- FORK：Linux 或 Mac（速度較快）
載入所需套件
匯出所需變數
將匯出的變數傳入具名參數
完成後關閉叢集

n_cores <- detectCores() - 2


cluster <- makeCluster(n_cores)




clusterEvalQ(cluster, library(crucial_package))

clusterExport(cluster, "variable_we_need")


parLapply(cluster, ls_inputs, our_function,
          named_argument = variable_we_need)


stopCluster(cluster)

一起來練習吧！

R 平行程式設計