Skip to content

Commit

Permalink
Implement DynamicDML
Browse files Browse the repository at this point in the history
* DynamicDML with all the OrthoLearner functionality
* API tests similar to the DML ones
  • Loading branch information
Miruna Oprescu committed Mar 31, 2021
1 parent 056fb30 commit 16dd7b7
Show file tree
Hide file tree
Showing 4 changed files with 926 additions and 2 deletions.
2 changes: 1 addition & 1 deletion econml/_cate_estimator.py
Original file line number Diff line number Diff line change
Expand Up @@ -563,7 +563,7 @@ def effect(self, X=None, *, T0, T1):
"""
Calculate the heterogeneous treatment effect :math:`\\tau(X, T0, T1)`.
The effect is calculatred between the two treatment points
The effect is calculated between the two treatment points
conditional on a vector of features on a set of m test samples :math:`\\{T0_i, T1_i, X_i\\}`.
Since this class assumes a linear effect, only the difference between T0ᵢ and T1ᵢ
matters for this computation.
Expand Down
7 changes: 6 additions & 1 deletion econml/dml/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -33,10 +33,14 @@
Orthogonal Statistical Learning.
ACM Conference on Learning Theory. `<https://arxiv.org/abs/1901.09036>`_
.. [dynamicdml] Greg Lewis and Vasilis Syrgkanis.
Double/Debiased Machine Learning for Dynamic Treatment Effects.
`<https://arxiv.org/abs/2002.07285>`_, 2021.
"""

from .dml import (DML, LinearDML, SparseLinearDML,
KernelDML, NonParamDML, ForestDML)
from .dynamic_dml import DynamicDML
from .causal_forest import CausalForestDML

__all__ = ["DML",
Expand All @@ -45,4 +49,5 @@
"KernelDML",
"NonParamDML",
"ForestDML",
"CausalForestDML", ]
"CausalForestDML",
"DynamicDML"]
Loading

0 comments on commit 16dd7b7

Please sign in to comment.