If only care about the higher subgroup (above cutoff), only need trt.est.high so set onlyhigh to be TRUE
Scores are adjusted to the opposite sign if higher.y == FALSE; scores stay the same if higher.y == TRUE;
this is because estcount.bilevel.subgroups() always takes the subgroup of the top highest adjusted scores,
and higher adjusted scores should always represent high responders of trt=1
estcount.bilevel.subgroups(
y,
x.cate,
x.ps,
time,
trt,
score,
higher.y,
prop,
onlyhigh,
ps.method = "glm",
minPS = 0.01,
maxPS = 0.99
)ate.est.high: estimated ATEs in the multiple bi-level subgroups that are in the higher-than-cutoff category;
vector of size equal to the length of prop; always returned
ate.est.low: estimated ATEs in the multiple bi-level subgroups that are in the lower-than-cutoff category;
vector of size equal to the length of prop; returned only when onlyhigh == TRUE
Observed outcome; vector of size n (observations)
Matrix of p.cate baseline covariates; dimension n by p.cate (covariates in the outcome model)
Matrix of p.ps baseline covariates (plus a leading column of 1 for the intercept);
dimension n by p.ps + 1 (covariates in the propensity score model plus intercept)
Log-transformed person-years of follow-up; vector of size n
Treatment received; vector of size n units with treatment coded as 0/1
Estimated log CATE scores for all n observations from one of the four methods
(boosting, naive Poisson, two regressions, contrast regression); vector of size n
A logical value indicating whether higher (TRUE) or lower (FALSE)
values of the outcome are more desirable. Default is TRUE.
Proportions corresponding to percentiles in the estimated log CATE scores that define subgroups to calculate ATE for;
vector of floats in `(0, 1]` (if onlyhigh=T) or in `(0, 1)` (if onlyhigh=F):
Each element of prop represents the high/low cutoff in each bi-level subgroup and the length of prop
is number of bi-level subgroups
Indicator of returning only the ATEs in the higher-than-cutoff category of the bi-level subgroups; boolean
A character value for the method to estimate the propensity score. Allowed values include one of:
'glm' for logistic regression with main effects only (default), or
'lasso' for a logistic regression with main effects and LASSO penalization on
two-way interactions (added to the model if interactions are not specified in ps.model).
Relevant only when ps.model has more than one variable.
A numerical value (in `[0, 1]`) below which estimated propensity scores should be
truncated. Default is 0.01.
A numerical value (in `(0, 1]`) above which estimated propensity scores should be
truncated. Must be strictly greater than minPS. Default is 0.99.