prepare_ae_specific function

Prepare datasets for AE specific analysis