Predicting customer churn with scikit-learn folowing yhat blog example