dte_adj

A Python Package for Estimating Distribution Treatment Effects

dte_adj is a Python package for estimating distribution treatment effects in randomized experiments. It provides APIs for conducting regression adjustment to estimate precise distribution functions, enabling deeper insights beyond average treatment effects through machine learning-enhanced estimation methods.

Estimator Types

The package provides several types of estimators for computing distribution treatment effects:

• Simple Randomization Estimators: For estimating distributional effects in simple randomized experiments where treatment assignment is independent of all covariates

• Covariate Adaptive Randomization Estimators: For estimating distributional effects under covariate-adaptive randomization (CAR) designs, including stratified block randomization and other adaptive schemes

• Local Distribution Estimators: For estimating local distribution treatment effects weighted by treatment propensity within strata

Theoretical Foundations

For theoretical foundations, see:

• Simple randomization: Byambadalai et al. (2024) [1]

• Covariate-adaptive randomization: Byambadalai et al. (2025) [2]

• Multi-task learning: Hirata et al. (2025) [3]

[1]

Byambadalai, U., Oka, T., & Yasui, S. (2024). Estimating Distributional Treatment Effects in Randomized Experiments: Machine Learning for Variance Reduction. arXiv preprint arXiv:2407.16037.

[2]

Byambadalai, U., Hirata, T., Oka, T., & Yasui, S. (2025). On Efficient Estimation of Distributional Treatment Effects under Covariate-Adaptive Randomization. arXiv preprint arXiv:2506.05945.

[3]

Hirata, T., Byambadalai, U., Oka, T., Yasui, S., & Uto, S. (2025). Efficient and Scalable Estimation of Distributional Treatment Effects with Multi-Task Neural Networks. arXiv preprint arXiv:2507.07738.

Contents:

• Installation Guide
• Get Started
• API Reference
• Contribution Guide

Indices and tables

• Index

• Module Index

• Search Page

License

MIT License