Add docstring

Quantco · Jul 10, 2024 · 2756dec · 2756dec
1 parent af0b641
commit 2756dec
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/metalearners/drlearner.py b/metalearners/drlearner.py
@@ -59,6 +59,9 @@ class DRLearner(_ConditionalAverageOutcomeMetaLearner):
 
         * ``"treatment_model"`` which estimates :math:`\mathbb{E}[Y(k) - Y(0) | X]`
 
+    If ``adaptive_clipping`` is set to ``True``, then the pseudo outcomes are computed using
+    adaptive propensity clipping described in section 4.1, equation *DR-Switch* of
+    `Mahajan et al. (2024) <https://arxiv.org/pdf/2211.01939>`_.
     """
 
     @classmethod