Mapping differential responses to cognitive training using machine learning