corentinchoisy/SPT
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Corrected bug in iptw function

Browse Source
This commit is contained in:
Corentin Choisy
2025-05-26 17:03:24 +02:00
parent fd6c1efe06
commit 3547d113f5
1 changed files with 2 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
R/iptw.R
View File

@ -34,13 +34,11 @@ iptw <- function(df=NULL,Y=NULL,X=NULL,target="ate") {
if (target == "ate") { if (target == "ate") {
psw <- df$TT/fitted(lr_out) + (1-df$TT)/(1-fitted(lr_out)) psw <- df$TT/fitted(lr_out) + (1-df$TT)/(1-fitted(lr_out))
} else if (target=="att") { } else if (target=="att") {
psw <- rep(NA,nrow(df)) psw <- fitted(lr_out)/(1-fitted(lr_out))
psw[df[,Y]==1] <- 1 psw[df[,Y]==1] <- 1
psw[df[,Y]==0] <- fitted(lr_out)/(1-fitted(lr_out))
} else if (target=="atu") { } else if (target=="atu") {
psw <- rep(NA,nrow(df)) psw <- (1-fitted(lr_out))/(fitted(lr_out))
psw[df[,Y]==0] <- 1 psw[df[,Y]==0] <- 1
psw[df[,Y]==1] <- (1-fitted(lr_out))/(fitted(lr_out))
} }
return(psw) return(psw)
} }