t_toy_data_simulation: Toy data simulation for T-learner

In xnie/rlearner: R-learner for Heterogeneous Treatment Effect Estimation

Description

Generates a toy dataset of size n that can be used to experiment with T-learners in this package. The generative process should be easy to learn with linear methods.

Usage

t_toy_data_simulation(n)

Arguments

n	the number of samples to draw from the distribution

Value

a list containing the covariate matrix, treatment vector, outcome vector, true propensity vector, true marginal outcome vector, true control potential outcome vector, true treated potential outcome vector, and true treatment effect vector, in that order.

Examples

data = t_toy_data_simulation(500) # draw a sample

xnie/rlearner documentation built on April 11, 2021, 12:49 a.m.