assign_subgroup_membership: Assign data to subgroups based on selected node
Description
Creates treatment recommendation flags based on identified subgroup
Usage
assign_subgroup_membership(data, best_subgroup, trees, X)
Value
Data frame with added predict.node and treat.recommend columns
Arguments
- data
Data frame. Original data
- best_subgroup
Data frame row. Selected subgroup information
- trees
List. Policy trees
- X
Matrix. Covariate matrix