
Doubly robust estimators of the coefficients in the two regression
Source:R/CATE_count.R
onearmglmcount.dr.RdDoubly robust estimators of the coefficients in the two regression
Arguments
- y
Observed outcome; vector of size
n- x.cate
Matrix of
pbaseline covariates; dimensionnbyp- time
Log-transformed person-years of follow-up; vector of size
n- trt
Treatment received; vector of size
nunits with treatment coded as 0/1- ps
Estimated propensity scores for all observations; vector of size
n- f.predictor
Initial prediction of the outcome (expected number of relapses for one unit of exposure time) conditioned on the covariates
xfor one treatment groupr;mu_r(x), step 1 in the two regression; vector of sizen