Update sklearn kaggle.py