Learn R Programming

forestsearch (version 0.1.0)

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