scripts_new_eff/sim_kara_YSmis1.R

In jlstiles/SDE_transport: SDEtransportsim

library(SDEtransportsim)

boots = 500
type = "YMmis"
load("func_forms1.RData")

if (type == "YMmis") {
  func_list = func_formsYM$func_listYMmis
  forms = func_formsYM$formsYSmis
}

if (type == "YSmis") {
  func_list = func_formsYS$func_listYSmis
  forms = func_formsYS$formsYSmis
}

if (type == "YZmis") {
  func_list = func_formsYZ$func_listYZmis
  forms = func_formsYZ$formsYSmis
}

system(paste0("mkdir -p ", paste0("results", type)))

sim_kara = function(n, forms, truth, B = boots) {
  
  data = gendata.SDEtransport(n, 
                              f_W = truth$f_W, 
                              f_S = truth$f_S, 
                              f_A = truth$f_A, 
                              f_Z = truth$f_Z, 
                              f_M = truth$f_M, 
                              f_Y = truth$f_Y)
  res = SDE_glm_seq(data, forms, RCT = 0.5, transport=T, 
                    pooled=T, gstar_S = 0, truth=truth, B = boots) 
  
  res_eff = SDE_glm_eff_seq(data, forms, RCT = 0.5, transport=T,
                            pooled=T, gstar_S = 0, truth=truth, B = boots)
  return(list(res= res, res_eff = res_eff))
}

library(parallel)

B = 1000
n=100

res100_YSmis = mclapply(1:B, FUN = function(x) sim_kara(n=100, forms=forms, truth=func_list, B = boots),
                        mc.cores = getOption("mc.cores", 20L))

save(res100_YSmis, func_list, forms, file = paste0("results", type, "/res100_YSmis.RData"))

B = 1000
n=500

res500_YSmis = mclapply(1:B, FUN = function(x) sim_kara(n=500, forms=forms, truth=func_list, B = boots),
                        mc.cores = getOption("mc.cores", 20L))

save(res500_YSmis, func_list, forms, file = paste0("results", type, "/res500_YSmis.RData"))

rm("res100_YSmis", "res500_YSmis")

B = 1000
n=5000

res5000_YSmis = mclapply(1:B, FUN = function(x) sim_kara(n=5000, forms=forms, truth=func_list, B = boots), 
                         mc.cores = getOption("mc.cores", 20L))

save(res5000_YSmis, func_list, forms, file = paste0("results", type, "/res5000_YSmis.RData"))