estimate_ite_bart {CRE}R Documentation

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

Description

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

Usage

estimate_ite_bart(y, z, X, learner_ps)

Arguments

y

An observed response vector.

z

A treatment vector.

X

A features matrix.

learner_ps

Method for the estimation of the propensity score.

Value

A list of ITE estimates.

Note

The number of samples and the number of burn are set by default equal to 500.


[Package CRE version 0.2.7 Index]