Function reference
Cross-Sectional Estimators
Quantile treatment effect estimators for cross-sectional data (qte.cross_sectional).
| estimate_unadjusted_effects | Estimate a quantile treatment effect by the unadjusted difference in quantiles. |
| estimate_ipw_effects | Estimate a quantile treatment effect by inverse probability weighting. |
| estimate_outcome_regression_effects | Estimate a quantile treatment effect by outcome regression. |
| estimate_aipw_effects | Estimate a quantile treatment effect by augmented inverse probability weighting. |
| QteResult | Result of a cross-sectional quantile treatment effect estimation. |
| CausalTarget | Causal estimand a cross-sectional estimator targets. |
| Estimator | Cross-sectional estimator that produced a result. |
Simulation
Simulation functions for the cross-sectional estimators (qte.cross_sectional.simulate).
| simulate_simple_data | Simulate a randomised experiment without covariates. |
| simulate_covariate_data | Simulate the fixed observational setting with covariates. |
| get_true_quantiles | Simulation-based potential-outcome quantiles and the quantile effect. |
| get_true_means | Simulation-based potential-outcome means and the average effect. |
Nonlinear Difference-in-Differences
Changes-in-changes and quantile difference-in-differences estimators for panel data (qte.nonlinear_did).
| estimate_nonlinear_did_for_panel | Estimate nonlinear difference-in-differences effects for panel data. |
| NonlinearDidResults | The four aggregations returned by estimate_nonlinear_did_for_panel. |
| NonlinearDidResult | Result for one aggregation of a nonlinear difference-in-differences estimation. |
| TrtGroupConfig | Lazy configuration for the treatment group column. |
| CounterfactualModel | Model used to construct the counterfactual distribution. |
| ControlGroup | Comparison group used for the counterfactual. |
| BasePeriod | Pre-treatment period the post-treatment effect is compared to. |
Data
Datasets bundled with py-qte (qte.datasets).
| load_lalonde | Load the LaLonde (1986) job training dataset. |
| load_mpdta | Load the minimum wage panel dataset (mpdta). |
| load_engel_with_weights | Load Engel (1857) food expenditure data with a sampling weights column. |
Bootstrap
Shared bootstrap configuration for the estimators (qte.bootstrap).
| BootstrapConfig | Configuration for the bootstrap. |