ATE_ntile: ATE_ntile

Description Usage Arguments Examples

Description

Estimate adjusted or nonadjusted ATE in subgroup.

Usage

1
ATE_ntile(model, Y, treatment, X, adjusted, estimation_dataset, treatment_dataset, ntilen, i)

Arguments

model
Y
treatment
X
adjusted
estimation_dataset
treatment_dataset
ntilen
i

Examples

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
##---- Should be DIRECTLY executable !! ----
##-- ==>  Define data, use random,
##--	or do  help(data=index)  for the standard data sets.

## The function is currently defined as
function (model, Y, treatment, X, adjusted, estimation_dataset,
    treatment_dataset, ntilen, i)
{
    pred_control <- data.frame(pred = predict(model, newdata = estimation_dataset),
        estimation_dataset)
    pred_treatment <- data.frame(pred = predict(model, newdata = treatment_dataset),
        treatment_dataset)
    estimate_result <- ATE_estimate(pred_control, pred_treatment,
        i, adjusted, Y, X, treatment, ntilen)
    dataset_count <- rbind(pred_control, pred_treatment) %>%
        mutate(nt = paste("ntile", ntile(pred, n = ntilen), sep = "_")) %>%
        group_by_(.dots = c("nt", treatment)) %>% summarise(count = n())
    return(estimate_result)
  }

yasui-salmon/stRatification documentation built on May 4, 2019, 2:31 p.m.