Predicting conditional average treatment effect (CATE) on a new policy based on the training over an old policy
make_cate(
model,
train_data,
test_data,
w,
x,
y,
family = gaussian(),
ntree = 100,
mtry = 2,
verbose = TRUE
)An object containing the estimated causal treatment effect results.
A model object used for estimation.
The training dataset.
The test dataset.
Set the treatment variable.
set Independent variables for the model.
Set the outcome variable.
The family type for the model (e.g., 'binomial').
Number of trees for the Random Forest model.
Number of variables to consider at each tree split in the Random Forest model.
Set TRUE to print the output on the console.
Athey, S., and Wager S. 2021. Policy Learning with Observational Data, Econometrica, 89, 1, 133–161.
Cerulli, G. 2021. Improving econometric prediction by machine learning, Applied Economics Letters, 28, 16, 1419-1425.
Cerulli, G. 2022. Optimal treatment assignment of a threshold-based policy: empirical protocol and related issues, Applied Economics Letters, DOI: 10.1080/13504851.2022.2032577.
Gareth, J., Witten, D., Hastie, D.T., Tibshirani, R. 2013. An Introduction to Statistical Learning : with Applications in R. New York, Springer.
Kitagawa, T., and A. Tetenov. 2018. Who Should Be Treated? Empirical Welfare Maximization Methods for Treatment Choice, Econometrica, 86, 2, 591–616.