estimate_ite_slearner {CRE}R Documentation

Estimate the Individual Treatment Effect (ITE) using S-Learner

Description

Estimates the Individual Treatment Effect using S-Learner given a response vector, a treatment vector, a features matrix and estimation model for the outcome.

Usage

estimate_ite_slearner(y, z, X, learner_y = "SL.xgboost")

Arguments

y

An observed response vector.

z

A treatment vector.

X

A features matrix.

learner_y

An estimation model for the outcome.

Value

A list of ITE estimates.


[Package CRE version 0.2.7 Index]