We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 64584b5 commit 0edfc12Copy full SHA for 0edfc12
R/double_ml_plr.R
@@ -524,8 +524,8 @@ DoubleMLPLR = R6Class("DoubleMLPLR",
524
d = self$data$data_model[[self$data$treat_col]]
525
y = self$data$data_model[[self$data$y_col]]
526
527
- psi_a = -(d - m_hat) * (d - m_hat)
528
- psi_b = (d - m_hat) * (y - l_hat)
+ psi_a = -(d - m_hat$preds) * (d - m_hat$preds)
+ psi_b = (d - m_hat$preds) * (y - l_hat$preds)
529
theta_initial = -mean(psi_b, na.rm = TRUE) / mean(psi_a, na.rm = TRUE)
530
531
data_aux = data.table(self$data$data_model,
0 commit comments