Replaced ShuffledStratifiedSplit by StratifiedKFold.