Improving Generalization in Game Agents with Data Augmentation in Imitation Learning