R/iptw_multiTrt.R

In JiayiJi/causal.multiple.treatments: Estimation of Causal Effects of Multiple Treatments with a Binary Outcome

iptw_multiTrt = function(y, trt_ind, psdat, estimand = "ATE", method) {
  if (estimand == "ATE") {
    iptw_est = iptw_multiTrt_ate(y, trt_ind, psdat, wt1, wt2, wt3, method)
  }
  if (estimand == "ATT") {
    iptw_est = iptw_multiTrt_att(y, trt_ind, psdat, wt12, wt13, method)
  }
  return(iptw_est)
}