https://cognitivehub.in/stratifiedkfold-vs-kfold-in-scikit-learn/