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