Data Augmentation for Automated Adaptive Rodent Training