Description Usage Arguments Details Value References Examples
Estimate the average treatment effect on the treated (ATT) via augmented inverse propensity weighting.
1 | aipw_rf_att(data, x, y, w, p, cf = TRUE, ...)
|
data |
a dataframe object containing the variables and values. |
x |
a list of character vectors specifying variables to be included in the model (columns in the data). If unspecified, then it is assumed to be all columns in the data besides y and w. |
y |
a character vector specifying the response variable. |
w |
a character vector specifying the treatment status. |
p |
a vector containing propensity score values. |
cf |
logical; if TRUE then includes confidence interval on ATT. |
... |
additional arguments to causal_forest. |
Computes an estimate of the ATT τ_T with a augmented inverse propensity score weighted estimate on just the treated group (see aipw_ate for more details).
a list of ATT, 95 percent confidence interval upperbound and lowerbound or just ATT, depending on user input of cf
.
Rotnitzky, Andrea, James M. Robins, and Daniel O. Scharfstein. 1998. “Semiparametric Regression for Repeated Outcomes with Nonignorable Nonresponse." Journal of the American Statistical Association. Vol. 93, No. 444, Dec. pgs. 1321-1339.
Cao, Weihua, Anastasios A. Tsiatis, and Marie Davidian. 2009. “Improving efficiency and robustness of the doubly robust estimator for a population mean with incomplete data". Biometrika. Vol. 96(3). pgs. 723–734.
1 2 3 4 5 | data("lalonde")
# calculate propensity scores
p <- propensity_score(lalonde, y = "re78", w = "treat", model = "logistic")
att <- aipw_rf_att(data = lalonde, y = "re78", w = "treat", p = p, num.trees = 100, mtry = 3)
|
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.