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.