simple_tree_data function

Generate simple tree data.

Generates covariates and potential outcomes of a synthetic dataset for a simple tree model.

simple_tree_data(
  A,
  K = 5,
  p = 10,
  noise_std = 1,
  split = 1.676,
  signal_strength = 1,
  noise_form = "normal"
)

Arguments

• A: Integer. Number of observations in the dataset. Must be a positive integer.
• K: Integer. Number of arms. Must be a positive integer.
• p: Integer. Number of covariates. Must be a positive integer.
• noise_std: Numeric. Standard deviation of the noise added to the potential outcomes. Must be a non-negative number.
• split: Numeric. Split point for creating treatment groups based on the covariates.
• signal_strength: Numeric. Strength of the signal in the potential outcomes.
• noise_form: Character. Distribution of the noise added to the potential outcomes. Can be either "normal" or "uniform".

Returns

A list containing the generated data (xs, ys, muxs) and the true potential outcome means (mus).

Examples

set.seed(123)
A <- 1000
K <- 4     # Number of treatment arms
p <- 10    # Number of covariates
synthetic_data <- simple_tree_data(A = A,
                                   K = K,
                                   p = p,
                                   noise_std = 1.0,
                                   split = 1.676,
                                   signal_strength = 1.0,
                                   noise_form = 'normal')