Learning how to Prune