-
Notifications
You must be signed in to change notification settings - Fork 1
/
run_para_boothal.R
38 lines (27 loc) · 1.09 KB
/
run_para_boothal.R
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
source(paste0("~/Repo/ph243/0_config.R"))
source(paste0("~/Repo/ph243/1_hal_undersmooth.R"))
source(paste0("~/Repo/ph243/2_estimation_function.R"))
source(paste0("~/Repo/ph243/6_hal_bootstrap.R"))
# devtools::load_all("/Users/haodongli/Repo/TMLEbootstrap/R")
# devtools::install_local("/Users/haodongli/Repo/TMLEbootstrap")
library(TMLEbootstrap)
registerDoFuture()
nCoresPerNode <- floor(as.numeric(Sys.getenv("SLURM_CPUS_ON_NODE"))/2)
nodeNames <-strsplit(Sys.getenv("SLURM_NODELIST"), ",")[[1]]
workers <- rep(nodeNames, each=nCoresPerNode)
cl = makeCluster(workers, type = "SOCK")
plan(cluster, workers = cl)
print(paste0("nCoresPerNode: ", nCoresPerNode))
# registerDoFuture()
# plan(multisession, workers=floor(2))
N_round = 500
n_sample = 500
# run simu
res <- run_simu(psi_true = 0.1153,
n_sample = n_sample,
N_round = N_round,
model_type = "para",
df_list = NULL)
output_filename <- paste0('~/Repo/ph243/results/', "para_boot_", n_sample, "_", Sys.Date(), '.csv')
write.csv(res, output_filename, row.names = FALSE)
stopCluster(cl)