stRatification is developed to analyze the Average Treatment Effect of subgroup that is build based on the prediction result. The original idea is introduced in this paper. Especially, this package is aim to reproduce the Table2 in the above paper. This package is still in developing, though you are able to reproduce the table 2 some degree.
You are able to install with install_github function in devtools package.
devtools::install_github("yasui-salmon/stRatification")
The original dataset is possible to download from here
Also you can find some definitions of the column in those data in here
You are able to find the example code and result here, please check this before you go for your analysis. Also the Rmd file of the above example is putted here
Abadie, Alberto, Matthew M. Chingos, and Martin R. West. Endogenous stratification in randomized experiments. No. w19742. National Bureau of Economic Research, 2013.