API Reference

This section provides comprehensive documentation for all classes and functions in the dte_adj package. The API is organized into logical groups based on functionality and use cases.

Overview

The dte_adj 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

• Utility Functions: Helper functions for confidence intervals and statistical computations

• Plotting Utilities: Visualization tools for treatment effects and distributions

For theoretical foundations, see Byambadalai et al. (2024) [1] for simple randomization and Byambadalai et al. (2025) [2] for covariate-adaptive randomization.

For multi-task learning approaches that train models for all locations simultaneously (using is_multi_task=True), see the neural network framework in [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.

Detailed Documentation

• Simple Randomization Estimators
  • SimpleDistributionEstimator
  • AdjustedDistributionEstimator
• Covariate Adaptive Randomization Estimators
  • SimpleStratifiedDistributionEstimator
  • AdjustedStratifiedDistributionEstimator
• Local Distribution Estimators
  • SimpleLocalDistributionEstimator
  • AdjustedLocalDistributionEstimator
• Plotting Utilities