vt1_rf {tehtuner}R Documentation

Estimate the CATE Using a Random Forest for Step 1 of Virtual Twins

Description

Estimate the CATE Using a Random Forest for Step 1 of Virtual Twins

Usage

vt1_rf(data, Trt, Y, ...)

Arguments

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.

...

additional arguments to rfsrc

Value

Estimated CATEs for each subject in data.

See Also

Other VT Step 1 functions: vt1_lasso(), vt1_mars(), vt1_super()


[Package tehtuner version 0.3.0 Index]