hte_plot

The function <code>hte_plot</code> 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.

Estimating heterogeneous treatment effects with tree-based machine
learning algorithms and visualizing estimated results in flexible and
presentation-ready ways. For more information, see Brand, Xu, Koch,
and Geraldo (2021) <doi:10.1177/0081175021993503>. Our current package
first started as a fork of the 'causalTree' package on 'GitHub' and we
greatly appreciate the authors for their extremely useful and free package.

Jiahui Xu

htetree

Causal Inference with Tree-Based Machine Learning Algorithms

Tanvi Shinkre

Jennie Brand

hte_plot function

<dl><dt>model</dt>
<dd>a tree model constructed by <code>hte_causalTree, hte_matchinleaves,
or hte_ipw</code>.</dd>
<dt>data</dt>
<dd>a data frame containing the variables in the model.</dd>
<dt>treatment_indicator</dt>
<dd>a character representing the column name
for the treatment variable in the causal setup.</dd>
<dt>outcomevariable</dt>
<dd>a character representing the column name
of the outcome variable.</dd>
<dt>propensity_score</dt>
<dd>a character representing the column name of
the propensity score.</dd>
<dt>plot.title</dt>
<dd>character representing the main title of the plot.</dd></dl>

Arguments

Visualize the Estimated Results — hte_plot

<dl>

<dt>model</dt>
<dd>a tree model constructed by <code>hte_causalTree, hte_matchinleaves,
or hte_ipw</code>.</dd>


<dt>data</dt>
<dd>a data frame containing the variables in the model.</dd>


<dt>treatment_indicator</dt>
<dd>a character representing the column name
for the treatment variable in the causal setup.</dd>


<dt>outcomevariable</dt>
<dd>a character representing the column name
of the outcome variable.</dd>


<dt>propensity_score</dt>
<dd>a character representing the column name of
the propensity score.</dd>


<dt>plot.title</dt>
<dd>character representing the main title of the plot.</dd>

</dl>

hte_plot: Visualize the Estimated Results

Description

Usage

Value

Arguments