changed inputs to predictions to folds not dataset