Skip to content

Commit

Permalink
Correct minor CausalForestDML documenation issue
Browse files Browse the repository at this point in the history
Signed-off-by: Keith Battocchi <kebatt@microsoft.com>
  • Loading branch information
kbattocchi committed Jun 28, 2024
1 parent 8dcdbb5 commit b20fcf2
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions econml/dml/causal_forest.py
Original file line number Diff line number Diff line change
Expand Up @@ -567,7 +567,7 @@ class CausalForestDML(_BaseDML):
ate_ : ndarray of shape (n_outcomes, n_treatments)
The average constant marginal treatment effect of each treatment for each outcome,
averaged over the training data and with a doubly robust correction. Available only
when `discrete_treatment=True` and `oob=True`.
when `discrete_treatment=True` and `drate=True`.
ate_stderr_ : ndarray of shape (n_outcomes, n_treatments)
The standard error of the `ate_` attribute.
feature_importances_ : ndarray of shape (n_features,)
Expand Down Expand Up @@ -997,7 +997,7 @@ def att__inference(self, *, T):
Inference results information for the `att_` attribute, which is the average
constant marginal treatment effect of each treatment for each outcome, averaged
over the training data treated with treatment T and with a doubly robust correction.
Available only when `discrete_treatment=True` and `oob=True`.
Available only when `discrete_treatment=True` and `drate=True`.
"""
return NormalInferenceResults(d_t=self._d_t[0] if self._d_t else 1,
d_y=self._d_y[0] if self._d_y else 1,
Expand Down

0 comments on commit b20fcf2

Please sign in to comment.