deploy: 1feb8aa

Author: TomeHirata

_images/distributional_effect.png

@@ -38,7 +38,7 @@ <h1>Source code for dte_adj.plot</h1><div class="highlight"><pre>
 
 
 <div class="viewcode-block" id="plot">
-<a class="viewcode-back" href="../../modules.html#dte_adj.plot.plot">[docs]</a>
+<a class="viewcode-back" href="../../api/plot.html#dte_adj.plot.plot">[docs]</a>
 <span class="k">def</span><span class="w"> </span><span class="nf">plot</span><span class="p">(</span>
     <span class="n">X</span><span class="p">:</span> <span class="n">np</span><span class="o">.</span><span class="n">ndarray</span><span class="p">,</span>
     <span class="n">means</span><span class="p">:</span> <span class="n">np</span><span class="o">.</span><span class="n">ndarray</span><span class="p">,</span>
@@ -135,8 +135,8 @@ <h1 class="logo"><a href="../../index.html">dte_adj</a></h1>
 <p class="caption" role="heading"><span class="caption-text">Contents:</span></p>
 <ul>
 <li class="toctree-l1"><a class="reference internal" href="../../installation.html">Installation Guide</a></li>
-<li class="toctree-l1"><a class="reference internal" href="../../get_started.html">Get Started with dte_adj</a></li>
-<li class="toctree-l1"><a class="reference internal" href="../../modules.html">Modules</a></li>
+<li class="toctree-l1"><a class="reference internal" href="../../get_started.html">Get Started</a></li>
+<li class="toctree-l1"><a class="reference internal" href="../../api_reference.html">API Reference</a></li>
 <li class="toctree-l1"><a class="reference internal" href="../../contributing.html">Contribution Guide</a></li>
 </ul>
 

_modules/dte_adj.html

@@ -63,8 +63,8 @@ <h1 class="logo"><a href="../index.html">dte_adj</a></h1>
 <p class="caption" role="heading"><span class="caption-text">Contents:</span></p>
 <ul>
 <li class="toctree-l1"><a class="reference internal" href="../installation.html">Installation Guide</a></li>
-<li class="toctree-l1"><a class="reference internal" href="../get_started.html">Get Started with dte_adj</a></li>
-<li class="toctree-l1"><a class="reference internal" href="../modules.html">Modules</a></li>
+<li class="toctree-l1"><a class="reference internal" href="../get_started.html">Get Started</a></li>
+<li class="toctree-l1"><a class="reference internal" href="../api_reference.html">API Reference</a></li>
 <li class="toctree-l1"><a class="reference internal" href="../contributing.html">Contribution Guide</a></li>
 </ul>
 

_modules/dte_adj/plot.html

@@ -0,0 +1,22 @@
+Local Distribution Estimators
+==============================
+
+This page documents local distribution treatment effect estimators that compute treatment effects weighted by treatment propensity within each stratum. These estimators are particularly useful for handling treatment assignment heterogeneity across strata.
+
+SimpleLocalDistributionEstimator
+--------------------------------
+
+.. autoclass:: dte_adj.SimpleLocalDistributionEstimator
+   :members:
+   :undoc-members:
+   :show-inheritance:
+   :no-index:
+
+AdjustedLocalDistributionEstimator
+----------------------------------
+
+.. autoclass:: dte_adj.AdjustedLocalDistributionEstimator
+   :members:
+   :undoc-members:
+   :show-inheritance:
+   :no-index:

_modules/index.html

@@ -0,0 +1,10 @@
+Plotting Utilities
+==================
+
+This page documents plotting utilities for visualizing distribution treatment effects, confidence intervals, and other results.
+
+.. automodule:: dte_adj.plot
+   :members:
+   :undoc-members:
+   :show-inheritance:
+   :no-index:

_sources/api/local.rst.txt

@@ -0,0 +1,30 @@
+Simple Randomization Estimators
+===============================
+
+This page documents estimators that work with simple randomized experimental designs where treatment assignment is completely randomized.
+
+These estimators leverage pre-treatment covariates through distributional regression frameworks to improve the precision of distributional treatment effect estimates. The key methodological contribution is using machine learning techniques for variance reduction while maintaining validity as long as nuisance components are reasonably well estimated.
+
+Byambadalai et al. (2024) [#simple2024]_ propose a regression adjustment method that incorporates covariates into distributional regression, enabling deeper insights beyond average treatment effects by estimating full distributional treatment effects in randomized experiments.
+
+.. [#simple2024] Byambadalai, U., Oka, T., & Yasui, S. (2024). Estimating Distributional Treatment Effects in Randomized Experiments: Machine Learning for Variance Reduction. arXiv preprint `arXiv:2407.16037 <https://arxiv.org/abs/2407.16037>`_.
+
+SimpleDistributionEstimator
+---------------------------
+
+.. autoclass:: dte_adj.SimpleDistributionEstimator
+   :members:
+   :inherited-members:
+   :undoc-members:
+   :show-inheritance:
+   :no-index:
+
+AdjustedDistributionEstimator
+-----------------------------
+
+.. autoclass:: dte_adj.AdjustedDistributionEstimator
+   :members:
+   :inherited-members:
+   :undoc-members:
+   :show-inheritance:
+   :no-index:

_sources/api/plot.rst.txt

@@ -0,0 +1,30 @@
+Covariate Adaptive Randomization Estimators
+===========================================
+
+This page documents estimators that work with stratified experimental designs, particularly for covariate-adaptive randomization (CAR) within strata.
+
+These estimators are designed to handle stratified block randomization where participants are grouped into strata based on baseline covariates before treatment assignment. The key methodological contribution is leveraging additional covariates beyond strata indicators using machine learning methods to enhance the precision of distributional treatment effect estimates.
+
+Byambadalai et al. (2025) [#car2025]_ propose a flexible distribution regression framework that achieves the semiparametric efficiency bound for distributional treatment effects under CAR, demonstrating that regression-adjusted estimators can optimally utilize covariate information in stratified designs.
+
+.. [#car2025] 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 <https://arxiv.org/abs/2506.05945>`_.
+
+SimpleStratifiedDistributionEstimator
+-------------------------------------
+
+.. autoclass:: dte_adj.SimpleStratifiedDistributionEstimator
+   :members:
+   :inherited-members:
+   :undoc-members:
+   :show-inheritance:
+   :no-index:
+
+AdjustedStratifiedDistributionEstimator
+---------------------------------------
+
+.. autoclass:: dte_adj.AdjustedStratifiedDistributionEstimator
+   :members:
+   :inherited-members:
+   :undoc-members:
+   :show-inheritance:
+   :no-index:

_sources/api/simple.rst.txt

@@ -0,0 +1,36 @@
+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) [#simple2024]_ for simple randomization and Byambadalai et al. (2025) [#car2025]_ 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 [#multitask2024]_.
+
+.. [#simple2024] Byambadalai, U., Oka, T., & Yasui, S. (2024). Estimating Distributional Treatment Effects in Randomized Experiments: Machine Learning for Variance Reduction. arXiv preprint `arXiv:2407.16037 <https://arxiv.org/abs/2407.16037>`_.
+
+.. [#car2025] 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 <https://arxiv.org/abs/2506.05945>`_.
+
+.. [#multitask2024] Byambadalai, U., Hirata, T., Oka, T., & Yasui, S. (2024). Efficient and Scalable Estimation of Distributional Treatment Effects with Multi-Task Neural Networks. arXiv preprint `arXiv:2507.07738 <https://arxiv.org/abs/2507.07738>`_.
+
+Detailed Documentation
+----------------------
+
+.. toctree::
+   :maxdepth: 2
+
+   api/simple
+   api/stratified
+   api/local
+   api/plot

_sources/api/stratified.rst.txt

@@ -1,5 +1,5 @@
-Get Started with dte_adj
-========================
+Get Started
+===========
 
 This page contains basic usage of dte_adj library.