estimate_ite_bart: Estimate the Individual Treatment Effect (ITE) using Bayesian...

View source: R/estimate_ite_bart.R

estimate_ite_bartR 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.


CRE documentation built on Oct. 19, 2024, 5:07 p.m.