HDCATE.set_condition_var: Set the conditional variable in CATE

In hdcate: Estimation of Conditional Average Treatment Effects with High-Dimensional Data

Set the conditional variable in CATE

Description

Set the conditional variable in CATE

Usage

HDCATE.set_condition_var(
  HDCATE_model,
  name = NA,
  min = NA,
  max = NA,
  step = NA
)

Arguments

HDCATE_model	an object created via HDCATE
name	name of the conditional variable
min	minimum value of the conditional variable for evaluation
max	maximum value of the conditional variable for evaluation
step	minimum distance between two evaluation points

Value

None. The HDCATE_model is ready to fit.

Examples

# get simulation data
n_obs <- 500  # Num of observations
n_var <- 100  # Num of observed variables
n_rel_var <- 4  # Num of relevant variables
data <- HDCATE.get_sim_data(n_obs, n_var, n_rel_var)
# conditional expectation model is misspecified
x_formula <- paste(paste0('X', c(2:n_var)), collapse ='+')
# propensity score model is misspecified
# x_formula <- paste(paste0('X', c(1:(n_var-1))), collapse ='+')

# create a new HDCATE model
model <- HDCATE(data=data, y_name='Y', d_name='D', x_formula=x_formula)

HDCATE.set_condition_var(model, 'X2', min=-1, max=1, step=0.01)

hdcate documentation built on Dec. 16, 2022, 5:19 p.m.