ATE_ntile: ATE_ntile

In yasui-salmon/stRatification: estimate the treatment effect in subgroup of dataset

Description

Estimate adjusted or nonadjusted ATE in subgroup.

Usage

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

##---- 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.