diff --git a/doc/conf.py b/doc/conf.py index 96c2aa37e..997000a89 100644 --- a/doc/conf.py +++ b/doc/conf.py @@ -18,8 +18,8 @@ # -- Project information ----------------------------------------------------- project = 'DoubleML' -copyright = '2021, Bach, P., Chernozhukov, V., Kurz, M. S., and Spindler, M.' -author = 'Bach, P., Chernozhukov, V., Kurz, M. S., and Spindler, M.' +copyright = '2021, Bach, P., Chernozhukov, V., Klaassen, S., Kurz, M. S., and Spindler, M.' +author = 'Bach, P., Chernozhukov, V., Klaassen, S., Kurz, M. S., and Spindler, M.' # The full version, including alpha/beta/rc tags release = '0.9.dev0' diff --git a/doubleml/datasets.py b/doubleml/datasets.py index a3be46ce7..74eceb0fc 100644 --- a/doubleml/datasets.py +++ b/doubleml/datasets.py @@ -1523,7 +1523,7 @@ def make_irm_data_discrete_treatments(n_obs=200, n_levels=3, linear=False, rando .. math:: - \\text{\\theta}(d) = 0.1 \\exp(d) + 10 \\sin(0.7 d) + 2 d - 0.2 d^2. + \\theta (d) = 0.1 \\exp(d) + 10 \\sin(0.7 d) + 2 d - 0.2 d^2. Based on the continous treatment, a discrete treatment :math:`D` is generated as with a baseline level of :math:`D=0` and additional levels based on the quantiles of :math:`D_{\\text{cont}}`. The number of levels @@ -1535,7 +1535,7 @@ def make_irm_data_discrete_treatments(n_obs=200, n_levels=3, linear=False, rando Y(0) &= 210 + 27.4 Z_1 + 13.7 (Z_2 + Z_3 + Z_4) + \\varepsilon_Y - Y(1) &= \\text{\\theta}(D_{\\text{cont}}) 1\\{D_{\\text{cont}} > 0\\} + Y(0), + Y(1) &= \\theta (D_{\\text{cont}}) 1\\{D_{\\text{cont}} > 0\\} + Y(0), where :math:`\\varepsilon_Y \\sim \\mathcal{N}(0,5)`. Further, the observed outcome is defined as @@ -1567,6 +1567,8 @@ def make_irm_data_discrete_treatments(n_obs=200, n_levels=3, linear=False, rando ------- res_dict : dictionary Dictionary with entries ``x``, ``y``, ``d`` and ``oracle_values``. + The oracle values contain the continuous treatment, the level bounds, the potential level, ITE + and the potential outcome without treatment. """ if random_state is not None: diff --git a/setup.py b/setup.py index 73f40c5c6..894fff48c 100644 --- a/setup.py +++ b/setup.py @@ -12,9 +12,9 @@ setup( name='DoubleML', version='0.9.dev0', - author='Bach, P., Chernozhukov, V., Kurz, M. S., and Spindler, M.', - maintainer='Malte S. Kurz', - maintainer_email='malte.simon.kurz@uni-hamburg.de', + author='Bach, P., Chernozhukov, V., Klaassen, S., Kurz, M. S., and Spindler, M.', + maintainer='Sven Klaassen', + maintainer_email='sven.klaassen@uni-hamburg.de', description='Double Machine Learning in Python', long_description=long_description, long_description_content_type='text/markdown',