Add comment to StratifiedDatasetCV