Machine-learning a virus assembly fitness landscape