vt2_classtree {tehtuner}

Estimate the CATE using a classification tree for Step 2

Description

Estimate the CATE using a classification tree for Step 2

Usage

vt2_classtree(z, data, Trt, Y, theta, threshold)

Arguments

z

a numeric vector of estimated CATEs from Step 1

data

a data frame containing a response, binary treatment indicators, and covariates.

Trt

a string specifying the name of the column of data contains the treatment indicators.

Y

a string specifying the name of the column of data contains the response.

theta

tree complexity parameter (cp)

threshold

for "step2 = 'classtree'" only. The value against which to test if the estimated individual treatment effect from Step 1 is higher (TRUE) or lower (FALSE).

Value

an object of class rpart. See rpart.object.

See Also

Other VT Step 2 functions: vt2_ctree(), vt2_lasso(), vt2_rtree()