Deep learning prediction of amplitude death