matchinleaves {htetree}

NN Matching in Leaves

Description

This intermediate function is used to adjust the heterogeneous treatment effect estimated in each leaf with NN matching.

Usage

matchinleaves(
  trainset = match_data,
  covariates = covariates,
  outcomevariable = outcomevariable,
  hte_effect_setup = hte_effect_setup,
  treatment_indicator,
  con.num = 1,
  ...
)

Arguments

trainset

a data frame only containing the variables used in the model and missings values are listwise deleted.

covariates

a vector of column names of all covariates (linear terms andpropensity score).

outcomevariable

a character representing the column name of the outcome variable.

hte_effect_setup

a empty list to store the adjusted treatment effect.

treatment_indicator

a character representing the column name of the treatment indicator.

con.num

a number indicating the number of units from control groups to be used in matching

...

further arguments passed to or from other methods.

Value

A list for summarizing the results after matching.