hte_plot {htetree}

Visualize the Estimated Results

Description

The function hte_plot takes a model created by causal tree, as well as the adjusted version, and plots the distribution of the outcome variable in treated and control groups in each leaf of the tree. This visualization aims to show how the predicted treatment effect changes with each split in the tree.

Usage

hte_plot(
  model,
  data,
  treatment_indicator = NULL,
  outcomevariable,
  propensity_score,
  plot.title = "Visualization of the Tree"
)

Arguments

model

a tree model constructed by hte_causalTree, hte_matchinleaves, or hte_ipw.

data

a data frame containing the variables in the model.

treatment_indicator

a character representing the column name for the treatment variable in the causal setup.

outcomevariable

a character representing the column name of the outcome variable.

propensity_score

a character representing the column name of the propensity score.

plot.title

character representing the main title of the plot.

Value

no return value