estimate_ite_bart

Estimates the Individual Treatment Effect using Bayesian Additive Regression
Trees given a response vector, a treatment vector, and a features matrix.

internal

Provides a new method for interpretable heterogeneous
treatment effects characterization in terms of decision rules
via an extensive exploration of heterogeneity patterns by an
ensemble-of-trees approach, enforcing high stability in the
discovery. It relies on a two-stage pseudo-outcome regression, and
it is supported by theoretical convergence guarantees. Bargagli-Stoffi,
F. J., Cadei, R., Lee, K., & Dominici, F. (2023) Causal rule ensemble:
Interpretable Discovery and Inference of Heterogeneous Treatment Effects.
arXiv preprint <doi:10.48550/arXiv.2009.09036>.

Falco Joannes Bargagli Stoffi

Interpretable Discovery and Inference of Heterogeneous Treatment
Effects

Naeem Khoshnevis

Daniela Maria Garcia

Riccardo Cadei

Kwonsang Lee

estimate_ite_bart function

<dl><dt>y</dt>
<dd>An observed response vector.</dd>
<dt>z</dt>
<dd>A treatment vector.</dd>
<dt>X</dt>
<dd>A features matrix.</dd>
<dt>learner_ps</dt>
<dd>Method for the estimation of the propensity score.</dd></dl>

Arguments

Estimate the Individual Treatment Effect (ITE) using Bayesian Additive
Regression Trees (BART) — estimate_ite_bart

<dl>

<dt>y</dt>
<dd>An observed response vector.</dd>


<dt>z</dt>
<dd>A treatment vector.</dd>


<dt>X</dt>
<dd>A features matrix.</dd>


<dt>learner_ps</dt>
<dd>Method for the estimation of the propensity score.</dd>

</dl>

estimate_ite_bart: Estimate the Individual Treatment Effect (ITE) using Bayesian Additive Regression Trees (BART)

Description

Usage

Value

Arguments