Fix for issue 151

nt-williams · Nov 4, 2024 · 9bec661 · 9bec661
1 parent 65e16c0
commit 9bec661
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/NEWS.md b/NEWS.md
@@ -8,6 +8,7 @@
 ### Bug Fixes
 
 -   Using fitted values from isotonic regression in `lmtp_survival()` instead of the original values (see issue \#149).
+-   Bootstrap TMLE uses cumulative density ratios (see issue \#151). 
 
 ### General
 

diff --git a/R/tmle2.R b/R/tmle2.R
@@ -1,4 +1,8 @@
 cf_tmle2 <- function(task, ratios, m_init, control) {
+  ratios <- matrix(t(apply(ratios, 1, cumprod)),
+                   nrow = nrow(ratios),
+                   ncol = ncol(ratios))
+
   psi <- estimate_tmle2(task$natural,
                         task$cens,
                         task$risk,