Estimate the CATE using a regression tree for Step 2
vt2_rtree(z, data, Trt, Y, theta)an object of class rpart. See
rpart.object.
a numeric vector of estimated CATEs from Step 1
a data frame containing a response, binary treatment indicators, and covariates.
a string specifying the name of the column of data
contains the treatment indicators.
a string specifying the name of the column of data
contains the response.
tree complexity parameter (cp)
Other VT Step 2 functions:
vt2_classtree(),
vt2_ctree(),
vt2_lasso()