Update scikit-learn cross validation approach