Update sklearn_training.py