I'm creating an AI model to generate density plots of crowds. When splitting the dataset into two,