optpolicy_functions.OptimalPolicy.allocate

OptimalPolicy.allocate(data_df, data_title='')

Allocate observations to treatment state.

Parameters

• data_df (DataFrame) – Input data with at least features or policy scores (depending on algorithm).

• data_title (String, optional) – This string is used as title in outputs. The default is ‘’.

Returns

• allocation_df (DataFrame) – data_df with optimal allocation appended.

• outpath (String) – Location of directory in which output is saved.