prepare_ae_specific_subgroup function

Prepare datasets for AE specific subgroup analysis