get_true_quantiles

cross_sectional.simulate.get_true_quantiles(
    qs: ArrayLike = MEDIAN,
    *,
    target: CausalTarget = CausalTarget.QTE,
    n_oracle: int = 1000000,
    effect_scale: float = 1.0,
    error_var: float = 1.0,
    seed: int = 0,
)

Simulation-based potential-outcome quantiles and the quantile effect.

The truth is recovered by simulating n_oracle units and taking empirical quantiles. With target=QTT the treated subsample is used.

Parameters

Name Type Description Default
qs array_like Probabilities in (0, 1). MEDIAN
target CausalTarget QTE for the population, QTT for the treated. CausalTarget.QTE
n_oracle int Size of the simulated population. 1000000
effect_scale float Treatment effect scale used in simulate_covariate_data. 1.0
error_var float Error variance used in simulate_covariate_data. 1.0
seed int Seed for the simulated population. 0

Returns

Name Type Description
polars.DataFrame Columns q, q_t, q_c and effect (q_t - q_c).