generate_training.py