Added execution time analysis to model_training script