Modelling Behavioural Diversity for Learning in Open-Ended Games