Skip to content

Commit b0404c0

Browse files
authored
Enable sample trimming in DRLearner (#1021)
This set of changes enables sample trimming in DRLearner (and subclasses), following Crump et al. (2009). The user can specify a fixed propensity threshold below which to drop observations, or can have the threshold automatically selected using the rule from the paper, which minimizes the expected variance of the estimates.
1 parent bed5a75 commit b0404c0

File tree

4 files changed

+654
-10
lines changed

4 files changed

+654
-10
lines changed

doc/spec/references.rst

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -135,3 +135,8 @@ References
135135
Syrgkanis, V., Lei, V., Oprescu, M., Hei, M., Battocchi, K., Lewis, G. (2019)
136136
Machine Learning Estimation of Heterogeneous Treatment Effects with Instruments
137137
URL https://arxiv.org/abs/1905.10176
138+
139+
.. [Crump2009]
140+
Crump, R. K., Hotz, V. J., Imbens, G. W., & Mitnik, O. A. (2009).
141+
Dealing with limited overlap in estimation of average treatment effects.
142+
Biometrika, 96(1), 187-199.

0 commit comments

Comments
 (0)